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