From 1547edddcac7d4f7883b60bc53808a750819e603 Mon Sep 17 00:00:00 2001
From: finn <finn@hillengass.de>
Date: Tue, 27 Feb 2024 20:26:30 +0100
Subject: [PATCH] Update file paths and variable names
---
src/one-shot/generate_annotations.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/one-shot/generate_annotations.py b/src/one-shot/generate_annotations.py
index be5250f..3746412 100644
--- a/src/one-shot/generate_annotations.py
+++ b/src/one-shot/generate_annotations.py
@@ -97,7 +97,7 @@ def annotate_dialogues(df: pd.DataFrame, api_url: str) -> pd.DataFrame:
pd.DataFrame: The DataFrame with added annotations.
"""
for index, row in tqdm(df.iterrows(), total=df.shape[0], desc="Annotating Dialogues"):
- prompt_original = llama_prompt_no_schema(row['Complete Dialogue'])
+ prompt_original = llama_prompt_no_schema(row['Original Dialogue'])
response_original = post_http_request(prompt_original, api_url, n=1, stream=False)
prompt_generated = llama_prompt_no_schema(row['utterances_joined'])
response_generated = post_http_request(prompt_generated, api_url, n=1, stream=False)
@@ -142,10 +142,10 @@ def main(args):
args: Command line arguments.
"""
api_url = f"http://{args.host}:{args.port}/generate"
- input_file = f"../../data/own_data/dialogues/{args.input_name}"
+ input_file = f"../../data/own_data/one-shot/dialogues/{args.input_name}"
df = pd.read_csv(input_file, sep=',', quoting=csv.QUOTE_NONE, escapechar='/')
df = annotate_dialogues(df, api_url)
- output_file = f"../../data/own_data/dialogues/{args.output_name}"
+ output_file = f"../../data/own_data/one-shot/dialogues/{args.output_name}"
df.to_csv(output_file, sep=',', index=False, quoting=csv.QUOTE_NONE, escapechar='/')
if __name__ == "__main__":
--
GitLab