Update app.py

app.py

@@ -14,7 +14,7 @@ CAPTION_MODELS = {
 captioner = pipeline(task="image-to-text",
                         model=CAPTION_MODELS['blip-base'],
                         max_new_tokens=30,
-                        device_map="auto", use_fast=True
+                        device_map="cpu", use_fast=True
                         )
 
 # Simple caption creation