Update app.py

app.py

@@ -64,20 +64,25 @@ def generate_api(model_str, prompt, seed=1):
     return None
 
 
-client = Client("Geek7/mdztxi2")
-result_path = client.predict(model_str=model_str,
-            prompt=prompt,
-            seed=seed,
-            api_name="/predict")
-            
-return result_path  
-if result_path:
-        # Send back the generated image file
-    return send_file(result_path, mimetype='image/png')
-else:
-    return jsonify({"error": "Failed to generate image."}), 500
+def predict():
+    data = request.get_json()
+    model_str = data['model_str']
+    prompt = data['prompt']
+    seed = data.get('seed', 1)
 
+    client = Client("Geek7/mdztxi2")
+    result_path = client.predict(
+        model_str=model_str,
+        prompt=prompt,
+        seed=seed,
+        api_name="/predict"
+    )
 
+    if result_path:
+        # Send back the generated image file
+        return send_file(result_path, mimetype='image/png')
+    else:
+        return jsonify({"error": "Failed to generate image."}), 500
 # Launch Gradio API without frontend
 iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file")
 iface.launch(show_api=True, share=True)