winain7788 commited on
Commit
ea3f038
·
1 Parent(s): 6f8a72b
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ sentiment_pipeline = pipeline("text-classification",
9
  device=device)
10
 
11
  async def get_sentiment(text):
12
- return sentiment_pipeline(text)
13
 
14
- demo = gr.Interface(fn=get_sentiment, inputs="text", outputs="json")
15
 
16
  demo.launch()
 
9
  device=device)
10
 
11
  async def get_sentiment(text):
12
+ return sentiment_pipeline([text])
13
 
14
+ demo = gr.Interface(fn=get_sentiment, inputs="text", outputs="list")
15
 
16
  demo.launch()