joshangngoching commited on
Commit
c97ff5c
·
verified ·
1 Parent(s): 7e7266c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  import cv2
6
  import matplotlib.cm as cm
7
 
8
- semantic_segmentation = pipeline("image-segmentation", "nvidia/segformer-b1-finetuned-cityscapes-1024-1024")
9
 
10
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png"])
11
 
 
5
  import cv2
6
  import matplotlib.cm as cm
7
 
8
+ semantic_segmentation = pipeline("image-segmentation", "nvidia/segformer-b5-finetuned-cityscapes-1024-1024")
9
 
10
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png"])
11