GirishKiran commited on
Commit
997cc72
1 Parent(s): 6bc975a

Upload ./app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -210,7 +210,7 @@ class SentimentAnalyser(object):
210
 
211
  # initalise the model
212
  def _init_model(self):
213
- modelLink = "bhadresh-savani/distilbert-base-uncased-emotion" #"SamLowe/roberta-base-go_emotions"
214
  self.tokenizer = AutoTokenizer.from_pretrained(modelLink)
215
  self.model = AutoModelForSequenceClassification.from_pretrained(modelLink)
216
  return
@@ -279,3 +279,4 @@ gradio.Interface(fn=predict_sentiment,
279
  description=desc,
280
  examples=exp,
281
  article=arti).launch(debug=True)
 
 
210
 
211
  # initalise the model
212
  def _init_model(self):
213
+ modelLink = "bhadresh-savani/distilbert-base-uncased-emotion"
214
  self.tokenizer = AutoTokenizer.from_pretrained(modelLink)
215
  self.model = AutoModelForSequenceClassification.from_pretrained(modelLink)
216
  return
 
279
  description=desc,
280
  examples=exp,
281
  article=arti).launch(debug=True)
282
+