winain7788 commited on
Commit
b79d7c8
·
1 Parent(s): 6aa1eb0
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,9 +6,9 @@ device = 0 if torch.cuda.is_available() else -1
6
 
7
  sentiment_pipeline = pipeline("sentiment-analysis", device=device)
8
 
9
- def process(text):
10
  return sentiment_pipeline(text)
11
 
12
- demo = gr.Interface(fn=process, inputs="text", outputs="json")
13
 
14
  demo.launch()
 
6
 
7
  sentiment_pipeline = pipeline("sentiment-analysis", device=device)
8
 
9
+ def get_sentiment(text):
10
  return sentiment_pipeline(text)
11
 
12
+ demo = gr.Interface(fn=get_sentiment, inputs="text", outputs="json")
13
 
14
  demo.launch()