Update app.py

app.py

@@ -35,7 +35,13 @@ def seed_everything(seed=42):
     torch.backends.cudnn.deterministic = True
 seed_everything(seed=CFG.seed) 
 
-
+tokenizer = AutoTokenizer.from_pretrained(CFG.model_name_or_path, return_tensors='pt')
+    
+if CFG.model == 't5':
+    model = AutoModelForSeq2SeqLM.from_pretrained(CFG.model_name_or_path).to(device)
+elif CFG.model == 'deberta':
+    model = EncoderDecoderModel.from_pretrained(CFG.model_name_or_path).to(device)
+    
 input_compound = CFG.input_data
 min_length = min(input_compound.find('CATALYST') - input_compound.find(':') - 10, 0)
 inp = tokenizer(input_compound, return_tensors='pt').to(device)