Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ import torchvision.transforms as transforms
|
|
35 |
|
36 |
import gradio as gr
|
37 |
|
38 |
-
model_name = "
|
39 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
40 |
def get_new_pallete(num_cls):
|
41 |
n = num_cls
|
|
|
35 |
|
36 |
import gradio as gr
|
37 |
|
38 |
+
model_name = "lseg_demo"
|
39 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
40 |
def get_new_pallete(num_cls):
|
41 |
n = num_cls
|