Update app.py

app.py

@@ -40,10 +40,10 @@ def image_preprocess(image):
     return image
 
 def generate_caption(processor, model, device, image):
-    #inputs = processor(image, return_tensors='pt').to(device)
-    #out = model.generate(**inputs, max_new_tokens=20)
-    #caption = processor.decode(out[0], skip_special_tokens=True)
-    caption="im here "
+    inputs =  image_processor (image, return_tensors='pt').to(device)
+    out = model.generate(**inputs, max_new_tokens=20)
+    caption = processor.decode(out[0], skip_special_tokens=True)
+    #caption="im here "
     return caption
 
 def main():