IS361Group4 commited on
Commit
9630d65
·
verified ·
1 Parent(s): 9d17b71

Update modules/sentiment.py

Browse files
Files changed (1) hide show
  1. modules/sentiment.py +1 -1
modules/sentiment.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
- classifier = pipeline("sentiment-analysis", model="cardiffnlp/twitter-xlm-roberta-base-sentiment")
5
 
6
  def sentiment_analysis(message, history):
7
  """
 
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
+ classifier = pipeline("sentiment-analysis", model="cardiffnlp/twitter-xlm-roberta-base-sentiment", framework="pt")
5
 
6
  def sentiment_analysis(message, history):
7
  """