Update app.py

app.py

@@ -50,16 +50,10 @@ def process_image(image):
             )
 
         # Decode and post-process the generated text
-        generated_text = processor.batch_decode(
+        return processor.batch_decode(
             generated_ids,
             skip_special_tokens=False
-        )[0]
-        caption = processor.post_process_generation(
-            generated_text,
-            task="<CAPTION>",
-            image_size=(image.width, image.height)
-        )
-        return caption
+        )[0].replace('</s>', '').replace('<s>', '').replace('<pad>', '').strip()
 
     except Exception as e:
         return f"Error processing image: {e}"