JackAILab commited on
Commit
dbeb86c
·
verified ·
1 Parent(s): 90165e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -42,6 +42,7 @@ bise_net = BiSeNet(n_classes = 19)
42
  bise_net.load_state_dict(torch.load(bise_net_cp_path, map_location="cpu")) # device fail
43
  bise_net.cuda()
44
 
 
45
  sys.path.append("./models/LLaVA")
46
  from llava.model.builder import load_pretrained_model
47
  from llava.mm_utils import get_model_name_from_path
 
42
  bise_net.load_state_dict(torch.load(bise_net_cp_path, map_location="cpu")) # device fail
43
  bise_net.cuda()
44
 
45
+ import sys
46
  sys.path.append("./models/LLaVA")
47
  from llava.model.builder import load_pretrained_model
48
  from llava.mm_utils import get_model_name_from_path