Update app.py
Browse files
app.py
CHANGED
@@ -54,16 +54,16 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
|
|
54 |
# Expose Gradio API
|
55 |
def generate_api(model_str, prompt, seed=1):
|
56 |
result = asyncio.run(infer(model_str, prompt, seed))
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
if result:
|
64 |
return result # Path to generated image
|
65 |
return None
|
66 |
|
67 |
# Launch Gradio API without frontend
|
68 |
-
iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file")
|
69 |
iface.launch(show_api=True, share=True)
|
|
|
54 |
# Expose Gradio API
|
55 |
def generate_api(model_str, prompt, seed=1):
|
56 |
result = asyncio.run(infer(model_str, prompt, seed))
|
57 |
+
# result = client.predict(
|
58 |
+
# model_str=model_str,
|
59 |
+
# prompt=prompt,
|
60 |
+
# seed=seed,
|
61 |
+
# api_name="/predict"
|
62 |
+
#)
|
63 |
if result:
|
64 |
return result # Path to generated image
|
65 |
return None
|
66 |
|
67 |
# Launch Gradio API without frontend
|
68 |
+
iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file", api_name="/predict")
|
69 |
iface.launch(show_api=True, share=True)
|