Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 113 |
-
|
| 114 |
-
|
| 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)
|