saadob12 commited on
Commit
b1dd4f4
·
1 Parent(s): 3e77d90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.0')
61
- self.model = AutoModelForSeq2SeqLM.from_pretrained('saadob12/t5_autochart_2.0').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)
 
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)