pentarosarium commited on
Commit
a7a5806
·
1 Parent(s): f602aaf
Files changed (1) hide show
  1. app.py +1 -1
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 = AutoModelForSeq2SeqM.from_pretrained(self.model_name)
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"