Update app.py

app.py

@@ -40,7 +40,7 @@ def generate_caption_and_image(image):
 
     
     # Generate caption
-    inputs = processor(raw_image, return_tensors="pt", padding=True, truncation=True, max_length=250)
+    inputs = processor(image, return_tensors="pt", padding=True, truncation=True, max_length=250)
     inputs = {key: val.to(device) for key, val in inputs.items()}
     out = model.generate(**inputs)
     caption = processor.decode(out[0], skip_special_tokens=True)