Spaces:

arpm01
/

financial-summarization

Runtime error

arpm01 commited on Mar 23, 2022

Commit

fc7c04c

1 Parent(s): 8142bba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,8 @@ extractor = AutoFeatureExtractor.from_pretrained("nickmuchi/segformer-b4-finetun
 model = SegformerForSemanticSegmentation.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
 pipe = pipeline(task="feature-extraction",
-                model=model)
 gr.Interface.load(pipe,
                     title="Segformer",

 model = SegformerForSemanticSegmentation.from_pretrained("nickmuchi/segformer-b4-finetuned-segments-sidewalk")
 pipe = pipeline(task="feature-extraction",
+                model=model,
+                extractor=extractor)
 gr.Interface.load(pipe,
                     title="Segformer",