Spaces:

rasyosef
/

roberta-sentiment-classifier

Sleeping

rasyosef commited on Jan 6

Commit

9bdd421

•

1 Parent(s): 62d0380

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,13 +1,13 @@
 import gradio as gr
 from transformers import pipeline
-roberta_sentiment = pipeline(
-    "text-classifier", model="rasyosef/roberta-base-finetuned-sst2"
 )
 def predict_sentiment(text):
-    return roberta_sentiment(text)
 with gr.Blocks() as demo:

 import gradio as gr
 from transformers import pipeline
+classifier = pipeline(
+    "text-classification", model="rasyosef/roberta-base-finetuned-sst2"
 )
 def predict_sentiment(text):
+    return classifier(text)[0]
 with gr.Blocks() as demo: