Pluto0616 commited on
Commit
5742893
·
verified ·
1 Parent(s): 32f1e23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -6,16 +6,15 @@ import torch.backends.cudnn as cudnn
6
  import gradio as gr
7
 
8
  os.system('git lfs install')
9
- os.system("pip install git+https://hf-mirror.com/Pluto0616/L2_InternVL")
10
- os.system('cd /home/user/app/L2_InternVL')
11
 
12
  from utils import load_json, init_logger
13
  from demo import ConversationalAgent, CustomTheme
14
 
15
- FOOD_EXAMPLES = "demo/food_for_demo.json"
16
  # MODEL_PATH = "/root/share/new_models/OpenGVLab/InternVL2-2B"
17
- MODEL_PATH = "work_dirs/internvl_v2_internlm2_2b_lora_finetune_food/lr35_ep10"
18
- OUTPUT_PATH = "./outputs"
19
 
20
  def setup_seeds():
21
  seed = 42
 
6
  import gradio as gr
7
 
8
  os.system('git lfs install')
9
+ os.system("git clone https://huggingface.co/Pluto0616/L2_InternVL")
 
10
 
11
  from utils import load_json, init_logger
12
  from demo import ConversationalAgent, CustomTheme
13
 
14
+ FOOD_EXAMPLES = "./L2_InternVL/demo/food_for_demo.json"
15
  # MODEL_PATH = "/root/share/new_models/OpenGVLab/InternVL2-2B"
16
+ MODEL_PATH = "./L2_InternVL/work_dirs/internvl_v2_internlm2_2b_lora_finetune_food/lr35_ep10"
17
+ OUTPUT_PATH = "./L2_InternVL/outputs"
18
 
19
  def setup_seeds():
20
  seed = 42