Update app.py

app.py

@@ -10,10 +10,15 @@ git_model = AutoModelForCausalLM.from_pretrained("microsoft/git-base-coco")
 blip_processor = AutoProcessor.from_pretrained("Salesfoce/blip-image-captioning-base")
 blip_model = BlipForConditionalGeneration.from_pretrained("Salesfoce/blip-image-captioning-base")
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+git_model.to(device)
+blip_model.to(device)
+
 def generate_caption(processor, model, image):
-    inputs = processor(image=image, return_tensors="pt")
+    inputs = processor(image=image, return_tensors="pt").to(device)
     
-    generated_ids = model.generate(pixel_values=pixel_values, max_length=50)
+    generated_ids = model.generate(pixel_values=inputs.pixel_values, max_length=50)
      
     generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]