Update app.py

app.py

@@ -6,17 +6,18 @@ import gradio as gr
 import PIL.Image
 import spaces
 import torch
-from transformers import BlipProcessor, BlipForConditionalGeneration
+from transformers import AutoProcessor, BlipForConditionalGeneration
 
 DESCRIPTION = "# Image Captioning with LongCap"
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model_id = "unography/blip-large-long-cap"
-processor = BlipProcessor.from_pretrained(model_id)
+processor = AutoProcessor.from_pretrained(model_id)
 model = BlipForConditionalGeneration.from_pretrained(model_id).to(device)
 
 
+
 def run(image: PIL.Image.Image) -> str:
     inputs = processor(images=image, return_tensors="pt").to(device)
     out = model.generate(pixel_values=inputs.pixel_values, max_length=300)