Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|