vignesha7 commited on
Commit
6e5f522
·
verified ·
1 Parent(s): 48ad800

Bug fix: updated imports

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ from peft import PeftModel
3
  from transformers import Qwen2VLForConditionalGeneration
4
  from transformers import AutoProcessor
5
  import gradio as gr
 
 
6
 
7
  #load the base model and finetuned adapter
8
  base_model = Qwen2VLForConditionalGeneration.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")
 
3
  from transformers import Qwen2VLForConditionalGeneration
4
  from transformers import AutoProcessor
5
  import gradio as gr
6
+ from transformers import Qwen2VLProcessor
7
+ from qwen_vl_utils import process_vision_info
8
 
9
  #load the base model and finetuned adapter
10
  base_model = Qwen2VLForConditionalGeneration.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")