Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig, TextStreamer
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig, TextStreamer, Qwen2VLForConditionalGeneration
 import torch
 from PIL import Image
 import re 
@@ -33,7 +33,7 @@ device_map = "auto"
 tokenizer = AutoTokenizer.from_pretrained(
         DEFAULT_CKPT_PATH, trust_remote_code=True,
     )
-model = AutoModelForCausalLM.from_pretrained(
+model = Qwen2VLForConditionalGeneration.from_pretrained(
         DEFAULT_CKPT_PATH,
         device_map=device_map,
         trust_remote_code=True,