Update app.py

app.py

@@ -1,10 +1,19 @@
-from flask import Flask, request
+from flask import request
 from diffusers import StableDiffusionPipeline
 import torch
-import os
+from fastapi import FastAPI, Response  
+from fastapi.middleware.cors import CORSMiddleware
 
 app = Flask(__name__)
 
+app.add_middleware(  # add the middleware
+    CORSMiddleware,
+    allow_credentials=True,  # allow credentials
+    allow_origins=["*"],  # allow all origins
+    allow_methods=["*"],  # allow all methods
+    allow_headers=["*"],  # allow all headers
+)
+
 
 model_id = "runwayml/stable-diffusion-v1-5"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
@@ -23,6 +32,3 @@ def generate_image():
     image_data = image.tobytes().hex()
 
     return {'image_data': image_data}
-
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000)