Update app.py

app.py

@@ -51,6 +51,12 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
 # Expose Gradio API
 def generate_api(model_str, prompt, seed=1):
     result = asyncio.run(infer(model_str, prompt, seed))
+    result = client.predict(
+            model_str=model_str,
+            prompt=prompt,
+            seed=seed,
+            api_name="/predict"
+        )
     if result:
         return result  # Path to generated image
     return None