entropy25 commited on
Commit
7f90963
·
verified ·
1 Parent(s): dd636d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -12
app.py CHANGED
@@ -109,23 +109,14 @@ class _MM9:
109
  def _load(self):
110
  try:
111
  self._dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
112
- # Using a fallback model since the base64 decoded path might not exist
113
- try:
114
- _mp = base64.b64decode("ZW50cm9weTI1L3NlbnRpbWVudGFuYWx5c2lz").decode()
115
- self._tok = BertTokenizer.from_pretrained(_mp)
116
- self._mdl = BertForSequenceClassification.from_pretrained(_mp)
117
- except:
118
- # Fallback to a publicly available model
119
- _mp = "nlptown/bert-base-multilingual-uncased-sentiment"
120
- self._tok = BertTokenizer.from_pretrained(_mp)
121
- self._mdl = BertForSequenceClassification.from_pretrained(_mp)
122
-
123
  self._mdl.to(self._dev)
124
  _log.info(f"Model loaded on {self._dev}")
125
  except Exception as e:
126
  _log.error(f"Model loading failed: {e}")
127
  raise
128
-
129
  class _TP8:
130
  @staticmethod
131
  @lru_cache(maxsize=_cfg._c5)
 
109
  def _load(self):
110
  try:
111
  self._dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
112
+ _mp = base64.b64decode("ZW50cm9weTI1L3NlbnRpbWVudGFuYWx5c2lz").decode()
113
+ self._tok = BertTokenizer.from_pretrained(_mp)
114
+ self._mdl = BertForSequenceClassification.from_pretrained(_mp)
 
 
 
 
 
 
 
 
115
  self._mdl.to(self._dev)
116
  _log.info(f"Model loaded on {self._dev}")
117
  except Exception as e:
118
  _log.error(f"Model loading failed: {e}")
119
  raise
 
120
  class _TP8:
121
  @staticmethod
122
  @lru_cache(maxsize=_cfg._c5)