Update run.py

authorship-verification-submission/run.py

@@ -7,19 +7,19 @@
 if __name__ == "__main__":
 
     # Load the data
-    tira = Client()
-    df = tira.pd.inputs(
+    tira_code = Client()
+    daf = tira_code.pd.inputs(
         "nlpbuw-fsu-sose-24", f"authorship-verification-validation-20240408-training"
     )
 
     # Load the model and make predictions
     model = load(Path(__file__).parent / "model.joblib")
-    predictions = model.predict(df["text"])
-    df["generated"] = predictions
-    df = df[["id", "generated"]]
+    predictions_model = model.predict(df["text"])
+    df["generated"] = predictions_model
+    daf = df[["id", "generated"]]
 
     # Save the predictions
-    output_directory = get_output_directory(str(Path(__file__).parent))
-    df.to_json(
-        Path(output_directory) / "predictions.jsonl", orient="records", lines=True
-    )
+    output = get_output(str(Path(__file__).parent))
+    daf.to_json(
+        Path(output) / "predictions.jsonl", orient="records", lines=True
+    )