Update app.py

app.py

@@ -9,7 +9,7 @@ from transformers import BlipProcessor, BlipForConditionalGeneration
 from openai import OpenAI
 # Load AI models
 caption_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-caption_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
+caption_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to("cpu")
 caption_model = torch.quantization.quantize_dynamic(caption_model, {torch.nn.Linear}, dtype=torch.qint8)
 client = OpenAI(
   api_key="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"