Spaces:

sankhyikii
/

segmentation_models

Runtime error

shyamgupta196 commited on Dec 30, 2022

Commit

3dc8b10

1 Parent(s): 2ea0e86

segformer extracter

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from transformers import AutoFeatureExtractor, SegformerForSemanticSegmentation
 from torch import nn
 import numpy as np
 import matplotlib.pyplot as plt
@@ -7,7 +7,7 @@ from PIL import Image
 import gradio as gr
 def seg(image):
-    feature_extractor = AutoFeatureExtractor.from_pretrained("nvidia/mit-b5")
     model = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b5")
     print(model)

+from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 from torch import nn
 import numpy as np
 import matplotlib.pyplot as plt
 import gradio as gr
 def seg(image):
+    feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/mit-b5")
     model = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b5")
     print(model)