Spaces:

aminaj
/

BrainBot

Running

aminaj commited on May 13

Commit

a347292

•

1 Parent(s): 5f40164

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -44,7 +44,7 @@ async def lifespan(app: FastAPI):
     image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
     yield
     # Delete all the temporary images
-    # utils.unlink_images("/images")
 ## FASTAPI APP
 # Initialize the FastAPI app

     image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
     yield
     # Delete all the temporary images
+    utils.unlink_images("/images")
 ## FASTAPI APP
 # Initialize the FastAPI app