Update app.py

app.py

@@ -1,17 +1,20 @@
 import gradio as gr
 import os
 
-# Use a pipeline as a high-level helper
-# Warning: Pipeline type "image-to-text" is no longer supported in transformers v5.
-# You must load the model directly (see below) or downgrade to v4.x with:
-# 'pip install "transformers<5.0.0'
-from transformers import pipeline
+from transformers import BlipForConditionalGeneration
+from transformers import AutoProcessor
+
+model = BlipForConditionalGeneration.from_pretrained(
+    "./models/Salesforce/blip-image-captioning-base")
+processor = AutoProcessor.from_pretrained(
+    "./models/Salesforce/blip-image-captioning-base")
 
-pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
 
 def caption(image):
-    out = pipe(image)
-    return out[0]['generated_text']
+    inputs = processor(image,return_tensors="pt")
+    out = model.generate(**inputs)
+
+    return processor.decode(out[0], skip_special_tokens=True)
 
 demo = gr.Interface(
     fn = caption,