Spaces:
Running
Running
Commit ·
a7a5806
1
Parent(s): f602aaf
3.60+
Browse files
app.py
CHANGED
|
@@ -71,7 +71,7 @@ class FallbackLLMSystem:
|
|
| 71 |
# Initialize MT5 model (multilingual T5)
|
| 72 |
self.model_name = "google/mt5-small"
|
| 73 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
| 74 |
-
self.model =
|
| 75 |
|
| 76 |
# Set device
|
| 77 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 71 |
# Initialize MT5 model (multilingual T5)
|
| 72 |
self.model_name = "google/mt5-small"
|
| 73 |
self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
|
| 74 |
+
self.model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
|
| 75 |
|
| 76 |
# Set device
|
| 77 |
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|