Spaces:

sankhyikii
/

segmentation_models

Runtime error

shyamgupta196 commited on Dec 30, 2022

Commit

2ea0e86

1 Parent(s): ff3b43f

remove resize

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,8 +7,6 @@ from PIL import Image
 import gradio as gr
 def seg(image):
-    ## first resize the image !!
-    image.resize((200,200))
     feature_extractor = AutoFeatureExtractor.from_pretrained("nvidia/mit-b5")
     model = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b5")
     print(model)

 import gradio as gr
 def seg(image):
     feature_extractor = AutoFeatureExtractor.from_pretrained("nvidia/mit-b5")
     model = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b5")
     print(model)