Update app.py

app.py

@@ -27,7 +27,7 @@ def generate_caption(processor, model, image, tokenizer=None):
     generated_ids = model.generate(pixel_values=inputs.pixel_values, max_length=50)
 
     if tokenizer is not None:
-        generated_ids = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
+        generated_caption = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
     else:
         generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]