Update app.py

app.py

@@ -85,11 +85,8 @@ def IMGEN():
     API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
     headers = {"Authorization": f"Bearer {key}"}
 
-    result = {
-        'response': requests.post(API_URL, headers=headers, json={"inputs": prompt,}).content.decode('utf-8', 'ignore')
-    }
 
-    return jsonify(result)
+    return requests.post(API_URL, headers=headers, json={"inputs": prompt,}).content
     
 
 @app.route('/divyanshpizza', methods=['GET'])