Update app/main.py

app/main.py

@@ -35,10 +35,10 @@ except Exception as e:
     print("❌ Firebase initialization failed:", e)
 
 # -------------------------------------------------
-# 📁 Directories
+# 📁 Directories (FIXED FOR HUGGINGFACE SPACES)
 # -------------------------------------------------
-UPLOAD_DIR = "uploads"
-RESULTS_DIR = "results"
+UPLOAD_DIR = "/tmp/uploads"
+RESULTS_DIR = "/tmp/results"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 os.makedirs(RESULTS_DIR, exist_ok=True)
 
@@ -54,16 +54,14 @@ model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)
 print("📦 Loading model weights...")
 state_dict = torch.load(model_path, map_location="cpu")
 
-# NOTE:
-# Replace this with your actual model class.
-# This is just example placeholder.
+# NOTE: Replace with real model architecture
 # from model import ColorizeNet
 # model = ColorizeNet()
 # model.load_state_dict(state_dict)
 # model.eval()
 
 def colorize_image(img: Image.Image):
-    """ Dummy colorizer (convert grayscale → fake color) """
+    """ Dummy colorizer (replace with real model.predict) """
     transform = transforms.ToTensor()
     tensor = transform(img.convert("L")).unsqueeze(0)
     tensor = tensor.repeat(1, 3, 1, 1)
@@ -98,7 +96,7 @@ async def upload_image(
     if not file.content_type.startswith("image/"):
         raise HTTPException(status_code=400, detail="Invalid file type")
 
-    image_id = str(uuid.uuid4()) + ".jpg"
+    image_id = f"{uuid.uuid4()}.jpg"
     file_path = os.path.join(UPLOAD_DIR, image_id)
 
     with open(file_path, "wb") as f:
@@ -128,7 +126,7 @@ async def colorize(
     img = Image.open(io.BytesIO(await file.read()))
     output_img = colorize_image(img)
 
-    result_id = str(uuid.uuid4()) + ".jpg"
+    result_id = f"{uuid.uuid4()}.jpg"
     output_path = os.path.join(RESULTS_DIR, result_id)
     output_img.save(output_path)