Update app.py

app.py

@@ -100,28 +100,6 @@ def inference_fn(test_loader, model, cfg):
 
     return np.concatenate(preds)
 
-def preprocess(df):
-    """
-    Preprocess the input DataFrame for training.
-
-    Args:
-        df (pd.DataFrame): Input DataFrame.
-        cfg (argparse.Namespace): Configuration object.
-
-    Returns:
-        pd.DataFrame: Preprocessed DataFrame.
-    """
-    df["input"] = (
-        "REACTANT:"
-        + df["REACTANT"]
-        + "REAGENT:"
-        + df["REAGENT"]
-        + "PRODUCT:"
-        + df["PRODUCT"]
-    )
-
-    return df
-
 
 class TestDataset(Dataset):
     """
@@ -203,9 +181,7 @@ if st.button('predict'):
         model = ReactionT5Yield.from_pretrained(CFG.model_name_or_path)
 
         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)
+            test_ds = pd.read_csv(CFG.uploaded_file)
         else:
             test_ds = pd.DataFrame.from_dict({"input": [CFG.data]}, orient="index").T