Update app.py

app.py

@@ -36,12 +36,11 @@ st.download_button(
 class CFG():
     uploaded_file = st.file_uploader("Choose a CSV file")
     data = st.text_area(display_text)
-    pretrained_model_name_or_path = 'sagawa/ZINC-t5'
     model = 't5'
-    model_name_or_path = './'
-    max_len = 512
+    model_name_or_path = 'sagawa/ReactionT5v2-yield'
+    max_len = 400
     batch_size = 5
-    fc_dropout = 0.1
+    fc_dropout = 0.0
     seed = 42
     num_workers=1
 
@@ -206,7 +205,7 @@ if st.button('predict'):
         if CFG.uploaded_file is not None:
             test_ds = pd.read_csv(CFG.data)
             if "input" not in test_ds.columns:
-                test_ds = preprocess(test_ds, CFG)
+                test_ds = preprocess(test_ds)
         else:
             test_ds = pd.DataFrame.from_dict({"input": [CFG.data]}, orient="index").T