Spaces:

Chand11
/

Image-captioner

Sleeping

Chand11 commited on Jun 22

Commit

538317c

verified ·

1 Parent(s): 188a06b

Update caption_app.py

Files changed (1) hide show

caption_app.py CHANGED Viewed

@@ -16,9 +16,10 @@ app = FastAPI()
 app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="templates")
-# Load BLIP model and processor
-processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
 @app.get("/", response_class=HTMLResponse)
 async def main(request: Request):

 app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="templates")
+# Load BLIP model and processor with local caching
+cache_dir = "./model_cache"
+processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base", cache_dir=cache_dir)
+model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base", cache_dir=cache_dir)
 @app.get("/", response_class=HTMLResponse)
 async def main(request: Request):