allyyy commited on
Commit
a4848fe
·
verified ·
1 Parent(s): ed2cfcb

Update model_evaluation.py

Browse files
Files changed (1) hide show
  1. model_evaluation.py +1 -1
model_evaluation.py CHANGED
@@ -6,7 +6,7 @@ from sentence_transformers import SentenceTransformer, util
6
  # Define paths and models
7
  filename = "output_topic_details.txt"
8
  retrieval_model_name = 'output/sentence-transformer-finetuned/' #using a prefine-tuned model
9
- gpt2_model_name = "gpt2"
10
  csv_file_path = "train_dataset.csv"
11
  output_csv_file_path = "updated_train_dataset.csv"
12
  val_csv_file_path = "val_dataset.csv"
 
6
  # Define paths and models
7
  filename = "output_topic_details.txt"
8
  retrieval_model_name = 'output/sentence-transformer-finetuned/' #using a prefine-tuned model
9
+ gpt2_model_name = "gpt-4o"
10
  csv_file_path = "train_dataset.csv"
11
  output_csv_file_path = "updated_train_dataset.csv"
12
  val_csv_file_path = "val_dataset.csv"