lazy loading

main.py

@@ -1,55 +1,28 @@
-# app/main.py
-
-from fastapi import FastAPI, UploadFile, File
+from fastapi import FastAPI
 from fastapi.responses import HTMLResponse
-from fastapi.staticfiles import StaticFiles
-import shutil
-from pathlib import Path
 import uvicorn
-
 from vit_captioning.generate import CaptionGenerator
 
 app = FastAPI()
-
-# Serve static files
-static_dir = Path(__file__).parent / "vit_captioning" / "static"
-app.mount("/static", StaticFiles(directory=static_dir), name="static")
-
-# ✅ Landing page at `/`
-# @app.get("/", response_class=HTMLResponse)
-# async def landing():
-#     return Path("vit_captioning/static/landing.html").read_text()
-
-@app.get("/")
-def read_root():
-    return {"message": "✅ Space is working"}
-
-# ✅ Captioning page at `/captioning`
-@app.get("/captioning", response_class=HTMLResponse)
-async def captioning():
-    return Path("vit_captioning/static/captioning/index.html").read_text()
-
-# ✅ Example: Project 2 placeholder
-@app.get("/project2", response_class=HTMLResponse)
-async def project2():
-    return "<h1>Coming Soon: Project 2</h1>"
-
-# ✅ Caption generation endpoint for captioning app
-# Keep the path consistent with your JS fetch()!
-caption_generator = CaptionGenerator(
-    model_type="CLIPEncoder",
-    checkpoint_path="./vit_captioning/artifacts/CLIPEncoder_40epochs_unfreeze12.pth",
-    quantized=False
-)
-
-@app.post("/generate")
-async def generate(file: UploadFile = File(...)):
-    temp_file = f"temp_{file.filename}"
-    with open(temp_file, "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
-
-    captions = caption_generator.generate_caption(temp_file)
-    return captions
-
-# if __name__ == "__main__":
-#     uvicorn.run(app, host="0.0.0.0", port=8000)
+caption_generator = None  # Lazy-load placeholder
+
+@app.on_event("startup")
+def startup_event():
+    global caption_generator
+    if caption_generator is None:
+        print("Loading CaptionGenerator...")
+        caption_generator = CaptionGenerator()
+
+@app.get("/", response_class=HTMLResponse)
+def root():
+    return "<h3>✅ Hugging Face Space is alive</h3>"
+
+# Example endpoint to trigger model
+@app.get("/caption")
+def caption():
+    if caption_generator is None:
+        return {"error": "Model not loaded"}
+    return {"result": "dummy caption"}  # Replace with real logic
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)