JackAILab commited on
Commit
457e1b4
Β·
verified Β·
1 Parent(s): e18909f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,9 +45,9 @@ bise_net.cuda()
45
 
46
  import sys
47
  sys.path.append("./models/LLaVA1.5/LLaVA/")
48
- from llava.model.builder import load_pretrained_model
49
- from llava.mm_utils import get_model_name_from_path
50
- from llava.eval.run_llava import eval_model
51
 
52
  # Load Llava for prompt enhancement
53
  llva_model_path = "liuhaotian/llava-v1.5-7b"
 
45
 
46
  import sys
47
  sys.path.append("./models/LLaVA1.5/LLaVA/")
48
+ from llava_infer.model.builder import load_pretrained_model
49
+ from llava_infer.mm_utils import get_model_name_from_path
50
+ from llava_infer.eval.run_llava import eval_model
51
 
52
  # Load Llava for prompt enhancement
53
  llva_model_path = "liuhaotian/llava-v1.5-7b"