Spaces:
Running
on
Zero
Running
on
Zero
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("
|
35 |
return "api ready"
|
36 |
except :
|
37 |
return "api not ready yet"
|
@@ -153,12 +153,12 @@ def get_riffusion(prompt):
|
|
153 |
return result[1]
|
154 |
|
155 |
def get_mustango(prompt):
|
156 |
-
client = Client("
|
157 |
result = client.predict(
|
158 |
-
prompt, # str in 'Prompt' Textbox component
|
159 |
-
200, # float (numeric value between 100 and 200) in 'Steps' Slider component
|
160 |
-
6, # float (numeric value between 1 and 10) in 'Guidance Scale' Slider component
|
161 |
-
|
162 |
)
|
163 |
print(result)
|
164 |
return result
|
|
|
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 |
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
|
160 |
+
guidance=6, # float (numeric value between 1 and 10) in 'Guidance Scale' Slider component
|
161 |
+
api_name="/predict"
|
162 |
)
|
163 |
print(result)
|
164 |
return result
|