Boyiliee commited on
Commit
115091d
1 Parent(s): 4df51ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ import torchvision.transforms as transforms
35
 
36
  import gradio as gr
37
 
38
- model_name = "convnext_xlarge_in22k"
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