Spaces:

cta2106
/

hawk-monitor

Runtime error

cta2106 commited on Oct 16, 2022

Commit

50c2eab

•

1 Parent(s): 13a626e

fixing

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ def predict_fn(text: str) -> str:
     tokenizer = LongformerTokenizer.from_pretrained("allenai/longformer-base-4096")
     p = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
-    return p(text)[1]["label"]
 gr.Interface(predict_fn, "textbox", "label").launch()

     tokenizer = LongformerTokenizer.from_pretrained("allenai/longformer-base-4096")
     p = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
+    return p(text)[0]["label"]
 gr.Interface(predict_fn, "textbox", "label").launch()