Update myapp.py

myapp.py

@@ -58,11 +58,11 @@ async def infer(client, prompt, seed=1, timeout=inference_timeout, model="prompt
 @myapp.route('/generate_api', methods=['POST'])
 def generate_api():
     data = request.get_json()
-
+    
     # Extract required fields from the request
     prompt = data.get('prompt', '')
     seed = data.get('seed', 1)
-    model_name = data.get('model', 'prompthero/openjourney-v4')  # Default to "prompthero/openjourney-v4" if not provided
+    model_name = data.get('model', 'prompthero/openjourney-v4')  # Default model
 
     if not prompt:
         return jsonify({"error": "Prompt is required"}), 400
@@ -74,7 +74,7 @@ def generate_api():
 
     try:
         # Create a generic InferenceClient for the model
-        client = InferenceClient(token=HF_TOKEN)  # Pass Hugging Face token if needed
+        client = InferenceClient(token=HF_TOKEN)
 
         # Call the async inference function
         result_path = asyncio.run(infer(client, prompt, seed, model=model))
@@ -83,6 +83,7 @@ def generate_api():
         else:
             return jsonify({"error": "Failed to generate image"}), 500
     except Exception as e:
+        print(f"Error in generate_api: {str(e)}")  # Log the error
         return jsonify({"error": str(e)}), 500
 
 # Add this block to make sure your app runs when called