Update app.py

app.py

@@ -51,11 +51,14 @@ def process_image(image):
 
         # Decode and post-process the generated text
         generated_text = processor.batch_decode(
-            generated_ids, skip_special_tokens=False
+            generated_ids,
+            skip_special_tokens=False
         )[0]
         caption = processor.post_process_generation(
-            generated_text, task="<CAPTION>", image_size=(image.width, image.height)
-        )[0]
+            generated_text,
+            task="<CAPTION>",
+            image_size=(image.width, image.height)
+        )
         return caption
 
     except Exception as e: