fixes

app.py

@@ -1,5 +1,5 @@
 import torch
-from transformers import AutoProcessor, AutoModelForCausalLM
+from transformers import AutoProcessor, AutoModelForVision2Seq
 from PIL import Image
 import gradio as gr
 
@@ -8,7 +8,7 @@ saved_folder_path = "model_folder"  # Replace with the path to your model folder
 
 # Load processor and model
 processor = AutoProcessor.from_pretrained(saved_folder_path)  # Processor (e.g., feature extractor + tokenizer)
-model = AutoModelForCausalLM.from_pretrained(saved_folder_path)  # Pre-trained model
+model = AutoModelForVision2Seq.from_pretrained(saved_folder_path)  # Pre-trained BLIP model
 model.eval()  # Set model to evaluation mode
 
 # Define the caption generation function
@@ -35,4 +35,4 @@ interface = gr.Interface(
 )
 
 # Launch the Gradio app
-interface.launch()
+interface.launch()

requirements.txt

@@ -1,8 +1,4 @@
-tensorflow==2.15.0
-gradio==5.3.0
-pandas==2.0.3
-Pillow==9.5.0
-torch==2.1.0
-nltk==3.8.1
-numpy==1.23.5
-transformers>=4.26.0
+torch
+transformers
+gradio
+pillow