GirishKiran commited on
Commit
e29ed4c
·
1 Parent(s): ef62575

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -201,9 +201,9 @@ class SentimentAnalyser(object):
201
  utility._ph()
202
  print(utility.fetch_system_info())
203
  utility._ph()
204
- # print(utility.fetch_gpu_info())
205
  utility._ph()
206
- # print(utility.fetch_host_ip())
207
  utility._ph()
208
  self._init_model()
209
  utility._login_hface()
@@ -211,7 +211,7 @@ class SentimentAnalyser(object):
211
 
212
  # initalise the model
213
  def _init_model(self):
214
- modelLink = "bhadresh-savani/distilbert-base-uncased-emotion" #"SamLowe/roberta-base-go_emotions"
215
  self.tokenizer = AutoTokenizer.from_pretrained(modelLink)
216
  self.model = AutoModelForSequenceClassification.from_pretrained(modelLink)
217
  self.whisper_model = whisper.load_model("small")
 
201
  utility._ph()
202
  print(utility.fetch_system_info())
203
  utility._ph()
204
+ print(utility.fetch_gpu_info())
205
  utility._ph()
206
+ print(utility.fetch_host_ip())
207
  utility._ph()
208
  self._init_model()
209
  utility._login_hface()
 
211
 
212
  # initalise the model
213
  def _init_model(self):
214
+ modelLink = "bhadresh-savani/distilbert-base-uncased-emotion"
215
  self.tokenizer = AutoTokenizer.from_pretrained(modelLink)
216
  self.model = AutoModelForSequenceClassification.from_pretrained(modelLink)
217
  self.whisper_model = whisper.load_model("small")