Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -57,8 +57,8 @@ class Model:
|
|
57 |
self.tokenizer = AutoTokenizer.from_pretrained('saadob12/t5_C2T_big')
|
58 |
self.model = AutoModelForSeq2SeqLM.from_pretrained('saadob12/t5_C2T_big').to(self.device)
|
59 |
elif mode.lower() == 'analytical':
|
60 |
-
self.tokenizer = AutoTokenizer.from_pretrained('saadob12/t5_autochart_2
|
61 |
-
self.model = AutoModelForSeq2SeqLM.from_pretrained('saadob12/t5_autochart_2
|
62 |
|
63 |
def generate(self):
|
64 |
tokens = self.tokenizer.encode(self.prefix + self.text, truncation=self.truncation, padding=self.padding, return_tensors='pt').to(self.device)
|
|
|
57 |
self.tokenizer = AutoTokenizer.from_pretrained('saadob12/t5_C2T_big')
|
58 |
self.model = AutoModelForSeq2SeqLM.from_pretrained('saadob12/t5_C2T_big').to(self.device)
|
59 |
elif mode.lower() == 'analytical':
|
60 |
+
self.tokenizer = AutoTokenizer.from_pretrained('saadob12/t5_autochart_2')
|
61 |
+
self.model = AutoModelForSeq2SeqLM.from_pretrained('saadob12/t5_autochart_2').to(self.device)
|
62 |
|
63 |
def generate(self):
|
64 |
tokens = self.tokenizer.encode(self.prefix + self.text, truncation=self.truncation, padding=self.padding, return_tensors='pt').to(self.device)
|