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
|