Update app.py

app.py

@@ -23,7 +23,7 @@ def set_page_config():
     )
 
 def initialize_model():
-    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    device = 'cpu'
     # load a fine-tuned image captioning model and corresponding tokenizer and image processor
     model = AutoModel.from_pretrained("sourabhbargi11/Caption_generator_model").to(device) 
     tokenizer = AutoTokenizer.from_pretrained("sourabhbargi11/Caption_generator_model")