Remove text subset

Milindi-Kodikara · Oct 7, 2024 · ab6c4cb · ab6c4cb
1 parent 4dc61cf
commit ab6c4cb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -77,7 +77,6 @@ def load_data_files(text_filepath, annotation_filepath):
         train_text, train_gold_standard_data = load_data_files(train_text_filepath, train_annotation_filepath)
         text, gold_standard_data = load_data_files(text_filepath, annotation_filepath)
 
-    text = text.head(2)
     prompts = load_prompts(prompt_filepath)
     model = get_model(model_id)