Update app.py
Browse files
app.py
CHANGED
|
@@ -65,7 +65,7 @@ class Translators:
|
|
| 65 |
def mitre(self):
|
| 66 |
from transformers import AutoModel, AutoTokenizer
|
| 67 |
tokenizer = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
|
| 68 |
-
model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True
|
| 69 |
# model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True)
|
| 70 |
# model.half() # recommended for GPU
|
| 71 |
model.eval()
|
|
|
|
| 65 |
def mitre(self):
|
| 66 |
from transformers import AutoModel, AutoTokenizer
|
| 67 |
tokenizer = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True, use_fast=False)
|
| 68 |
+
model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True).to(self.device)
|
| 69 |
# model = AutoModel.from_pretrained(self.model_name, trust_remote_code=True)
|
| 70 |
# model.half() # recommended for GPU
|
| 71 |
model.eval()
|