fffiloni commited on
Commit
05d54c4
·
verified ·
1 Parent(s): fa2adba

Update Mustango API

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ def check_api(model_name):
31
  return "api not ready yet"
32
  elif model_name == "Mustango":
33
  try :
34
- client = Client("fffiloni/mustango-API", hf_token=hf_token)
35
  return "api ready"
36
  except :
37
  return "api not ready yet"
@@ -153,7 +153,7 @@ def get_riffusion(prompt):
153
  return result[1]
154
 
155
  def get_mustango(prompt):
156
- client = Client("fffiloni/mustango-API", hf_token=hf_token)
157
  result = client.predict(
158
  prompt=prompt, # str in 'Prompt' Textbox component
159
  steps=200, # float (numeric value between 100 and 200) in 'Steps' Slider component
 
31
  return "api not ready yet"
32
  elif model_name == "Mustango":
33
  try :
34
+ client = Client("fffiloni/mustango-API-2", hf_token=hf_token)
35
  return "api ready"
36
  except :
37
  return "api not ready yet"
 
153
  return result[1]
154
 
155
  def get_mustango(prompt):
156
+ client = Client("fffiloni/mustango-API-2", hf_token=hf_token)
157
  result = client.predict(
158
  prompt=prompt, # str in 'Prompt' Textbox component
159
  steps=200, # float (numeric value between 100 and 200) in 'Steps' Slider component