...3 app update

app.py

@@ -1,27 +1,23 @@
-from fastapi import FastAPI
-from fastapi import Request
+from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 from diffusers import DiffusionPipeline
 import torch
 import uuid
-from PIL import Image
 import os
+from PIL import Image
 from fastapi.staticfiles import StaticFiles
 
 app = FastAPI()
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["http://localhost:5173",
-                   "https://react-portfolio-git-main-pacicaps-projects.vercel.app",
-                   "https://react-portfolio-ij8ifou62-pacicaps-projects.vercel.app"],
+    allow_origins=["*"],  # Accept from all for now
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# Remote Hugging Face model IDs
 hf_model_ids = {
     "model1": "Pacicap/FineTuned_claude_StableDiffussion_2_1",
     "model2": "Pacicap/FineTuned_gpt4o_StableDiffussion_2_1"
@@ -31,7 +27,7 @@ loaded_models = {}
 
 class PromptInput(BaseModel):
     prompt: str
-    model: str  # should be "model1" or "model2"
+    model: str
 
 @app.post("/generate")
 def generate(data: PromptInput, request: Request):
@@ -43,8 +39,10 @@ def generate(data: PromptInput, request: Request):
     model_id = hf_model_ids[model_key]
 
     if model_key not in loaded_models:
-        device = "cuda" if torch.cuda.is_available() else "cpu"
-        pipe = DiffusionPipeline.from_pretrained(model_id).to(device)
+        pipe = DiffusionPipeline.from_pretrained(
+            model_id,
+            torch_dtype=torch.float32
+        ).to("cpu")  # CPU-safe for Spaces
         loaded_models[model_key] = pipe
     else:
         pipe = loaded_models[model_key]
@@ -56,10 +54,8 @@ def generate(data: PromptInput, request: Request):
     filepath = os.path.join("generated", filename)
     image.save(filepath)
 
-    image_url = f"{request.base_url}generated/{filename}"
-    return {"url": image_url}
-
-    #return {"url": f"http://localhost:8000/generated/{filename}"}
+    return {
+        "url": f"{request.base_url}generated/{filename}"
+    }
 
-# Serve images
 app.mount("/generated", StaticFiles(directory="generated"), name="generated")