Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 # import prometheus_client as prom
 # import pandas as pd
 # import uvicorn
-from transformers import VisionEncoderDecoderModel,pipeline, AutoTokenizer, AutoImageProcessor
+from transformers import VisionEncoderDecoderModel,pipeline, ViTImageProcessor, AutoTokenizer
 import torch
 
 
@@ -38,7 +38,7 @@ import torch
 
 #     f1_metric.set(f1)
 
-vitgpt_processor = AutoImageProcessor.from_pretrained("model")
+vitgpt_processor = ViTImageProcessor.from_pretrained("model")
 vitgpt_model = VisionEncoderDecoderModel.from_pretrained("model")
 vitgpt_tokenizer = AutoTokenizer.from_pretrained("model")