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"
|