fffiloni commited on
Commit
0fc79c2
·
verified ·
1 Parent(s): cdd761d

re-enable audiogen

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -108,12 +108,13 @@ def get_audioldm(prompt):
108
 
109
  def get_audiogen(prompt):
110
  try:
111
- client = Client("https://fffiloni-audiogen.hf.space/")
112
  result = client.predict(
113
- prompt,
114
- 10,
115
- api_name="/infer"
116
  )
 
117
  return result
118
  except:
119
  raise gr.Error("AudioGen space API is not ready, please try again in few minutes ")
@@ -249,7 +250,7 @@ with gr.Blocks(css=css) as demo:
249
  chosen_model = gr.Dropdown(label="Choose a model", choices=[
250
  #"MAGNet",
251
  "AudioLDM-2",
252
- #"AudioGen",
253
  "Tango",
254
  #"Tango 2",
255
  "Stable Audio Open",
 
108
 
109
  def get_audiogen(prompt):
110
  try:
111
+ client = Client("fffiloni/audiogen", hf_token=hf_token)
112
  result = client.predict(
113
+ prompt=prompt,
114
+ duration=10,
115
+ api_name="/infer"
116
  )
117
+ print(result)
118
  return result
119
  except:
120
  raise gr.Error("AudioGen space API is not ready, please try again in few minutes ")
 
250
  chosen_model = gr.Dropdown(label="Choose a model", choices=[
251
  #"MAGNet",
252
  "AudioLDM-2",
253
+ "AudioGen",
254
  "Tango",
255
  #"Tango 2",
256
  "Stable Audio Open",