Update app.py

app.py

@@ -18,11 +18,11 @@ def predict(model_version, dataset_name, input_type, file=None, smiles=None):
     if model_version == "Vanilla Chemprop":
         model_dir = "chemprop"
         model_path = f"model_weight/{dataset_name}/best_unbalanced.pt"
-        script_path = "chemprop/chemprop/cli/predict"
+        script_path = "chemprop/chemprop/cli/predict.py"
     else:
         model_dir = "chemprop_updated"
         model_path = f"model_weight/{dataset_name}/best_bert_fusion.pt"
-        script_path = "chemprop-updated/chemprop_updated/cli/predict"
+        script_path = "chemprop-updated/chemprop_updated/cli/predict.py"
 
     # Prepare input file
     if input_type == "Upload CSV":