Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import torch
-from transformers import LlavaProcessor, LlavaForConditionalGeneration
+from transformers import AutoProcessor, LlavaForConditionalGeneration
 from PIL import Image
 
 # 加载模型
 model_id = "Yanqing0327/LLaVA-project"
-processor = LlavaProcessor.from_pretrained(model_id)
+processor = AutoProcessor.from_pretrained(model_id)
 model = LlavaForConditionalGeneration.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float16)
 
 def llava_infer(image, text):