Update app.py

app.py

@@ -72,11 +72,11 @@ class BLIPImageCaptioning:
 
     def generate_caption(self, image):
         try:
-            model = BlipForConditionalGeneration.from_pretrained(self.blip_model_name).to("cuda")
+            model = BlipForConditionalGeneration.from_pretrained(self.blip_model_name)
             processor = BlipProcessor.from_pretrained(self.blip_model_name)
             
             raw_image = self.preprocess_image(image)
-            inputs = processor(raw_image, return_tensors="pt").to("cuda")
+            inputs = processor(raw_image, return_tensors="pt")
             out = model.generate(**inputs)
             unconditional_caption = processor.decode(out[0], skip_special_tokens=True)
 
@@ -176,7 +176,7 @@ def preprocessing_interface(original_image, brightness_slider, contrast_slider,
         return gr.Error(f"Error in preprocessing: {str(e)}")
 
 def generate_captions(images):
-    blip_model = BlipForConditionalGeneration.from_pretrained(blip_model_name).to("cuda")
+    blip_model = BlipForConditionalGeneration.from_pretrained(blip_model_name)
     blip_processor = BlipProcessor.from_pretrained(blip_model_name)
     
     return [blip_model_instance.generate_caption(image) for image in images]