SiyouLi commited on
Commit
d3bb657
·
verified ·
1 Parent(s): 51b859a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import os
11
  import sys
12
  from typing import Optional, List, Tuple
13
  os.system('pip install torch==2.9.0 torchvision==0.24.0 torchaudio==2.9.0 --index-url https://download.pytorch.org/whl/cpu -U')
14
- os.system('pip install transformers==4.57.1 av qwen-vl-utils sentencepiece bitsandbytes ')
15
  import gradio as gr
16
  import torch
17
  from transformers import AutoModelForCausalLM, AutoProcessor
@@ -23,7 +23,7 @@ PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
23
  if PROJECT_ROOT not in sys.path:
24
  sys.path.insert(0, PROJECT_ROOT)
25
 
26
- from src.utils.qwen_vision_process import process_vision_info
27
 
28
 
29
  DEFAULT_MODEL_ID = os.environ.get("QTSPLUS_MODEL_ID", "AlpachinoNLP/QTSplus-3B")
 
11
  import sys
12
  from typing import Optional, List, Tuple
13
  os.system('pip install torch==2.9.0 torchvision==0.24.0 torchaudio==2.9.0 --index-url https://download.pytorch.org/whl/cpu -U')
14
+ os.system('pip install transformers==4.57.1 av qwen-vl-utils sentencepiece bitsandbytes -U')
15
  import gradio as gr
16
  import torch
17
  from transformers import AutoModelForCausalLM, AutoProcessor
 
23
  if PROJECT_ROOT not in sys.path:
24
  sys.path.insert(0, PROJECT_ROOT)
25
 
26
+ from qwen_vl_utils import process_vision_info
27
 
28
 
29
  DEFAULT_MODEL_ID = os.environ.get("QTSPLUS_MODEL_ID", "AlpachinoNLP/QTSplus-3B")