Update app.py

app.py

@@ -275,6 +275,7 @@ if run:
             else None,
             seed=int(seed),
             batch_size=int(batch_size),
+            debug=False
         )
 
         seed_everything(seed=CFG.seed)
@@ -294,8 +295,10 @@ if run:
         # Prepare data
         file_bytes = uploaded.getvalue()
         input_df = parse_csv_from_bytes(file_bytes)
-        # input_df = pd.read_csv(uploaded)
-        input_df = preprocess_df(input_df, drop_duplicates=False)
+        if task != "yield prediction":
+            input_df = preprocess_df(input_df, drop_duplicates=False)
+        else:
+            input_df = preprocess_df(input_df, cfg=CFG,drop_duplicates=False)
 
         # Dataset & loader
         dataset = ReactionT5Dataset(CFG, input_df)