Spaces:

arpm01
/

financial-summarization

Runtime error

arpm01 commited on Mar 23, 2022

Commit

780f382

1 Parent(s): a93815b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ model = SegformerForSemanticSegmentation.from_pretrained("nickmuchi/segformer-b4
 pipe = pipeline(task="image-segmentation",
                 model=model,
-                extractor=extractor)
 gr.Interface.load(pipe,
                     title="Segformer",

 pipe = pipeline(task="image-segmentation",
                 model=model,
+                feature_extractor=extractor)
 gr.Interface.load(pipe,
                     title="Segformer",