Spaces:
Sleeping
Sleeping
Update Mustango API
Browse files
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
|