Update app.py

app.py

@@ -89,12 +89,12 @@ class RegressionModel(nn.Module):
         else:
             self.config = torch.load(config_path)
         if pretrained:
-            if 't5' in cfg.pretrained_model_name_or_path:
+            if 't5' in cfg.model:
                 self.model = T5EncoderModel.from_pretrained(CFG.pretrained_model_name_or_path)
             else:
                 self.model = AutoModel.from_pretrained(CFG.pretrained_model_name_or_path)
         else:
-            if 't5' in cfg.model_name_or_path:
+            if 't5' in cfg.model:
                 self.model = T5EncoderModel.from_pretrained('sagawa/ZINC-t5')
             else:
                 self.model = AutoModel.from_config(self.config)