Update app.py

app.py

@@ -86,7 +86,7 @@ def IMGEN():
     headers = {"Authorization": f"Bearer {key}"}
 
     result = {
-        'response': requests.post(API_URL, headers=headers, json=payload).content
+        'response': requests.post(API_URL, headers=headers, json={"inputs": prompt,}).content
     }
 
     return jsonify(result)