Update app.py

app.py

@@ -55,13 +55,13 @@ async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
     return None
 
 # Expose Gradio API
-def generate_api(model_str, prompt, seed=1):
+#def generate_api(model_str, prompt, seed=1):
     result = asyncio.run(infer(model_str, prompt, seed))
     if result:
         return result  # Path to generated image
     return None
 
-@app.route('/predict', methods=['POST'])
+@app.route('/predict2', methods=['POST'])
 def async_infer_api():
     data = request.get_json()
     model_str = data['model_str']
@@ -81,5 +81,5 @@ def async_infer_api():
 
 
 # Launch Gradio API without frontend
-iface = gr.Interface(fn=generate_api, inputs=["text", "text", "number"], outputs="file")
+iface = gr.Interface(fn=async_infer_api , inputs=["text", "text", "number"], outputs="file")
 iface.launch(show_api=True, share=True)