taratrankennedy commited on
Commit
3f325ac
1 Parent(s): 78247ea

Update model_evaluation.py

Browse files
Files changed (1) hide show
  1. model_evaluation.py +1 -1
model_evaluation.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline, GPT2Tokenizer
4
  from sentence_transformers import SentenceTransformer, util
5
 
6
  # Define paths and models
7
- filename = "output_country_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"
 
4
  from sentence_transformers import SentenceTransformer, util
5
 
6
  # Define paths and models
7
+ filename = "output_chess_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"