Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from PIL import Image
-from transformers import AutoProcessor, Blip2ForConditionalGeneration
+from transformers import AutoProcessor, BlipForConditionalGeneration
 import os
-processor = AutoProcessor.from_pretrained("shadowlilac/visor-v2", token=os.getenv('HUGGING_FACE_HUB_TOKEN'))
-model = Blip2ForConditionalGeneration.from_pretrained("shadowlilac/visor-v2", ignore_mismatched_sizes=True, token=os.getenv('HUGGING_FACE_HUB_TOKEN'))
+processor = AutoProcessor.from_pretrained("shadowlilac/visor", token=os.getenv('HUGGING_FACE_HUB_TOKEN'))
+model = BlipForConditionalGeneration.from_pretrained("shadowlilac/visor", ignore_mismatched_sizes=True, token=os.getenv('HUGGING_FACE_HUB_TOKEN'))
 
 def generate_caption(image):
     raw_image = Image.fromarray(image)