diff --git a/src/model/translate.py b/src/model/translate.py index 25089cd..5163999 100644 --- a/src/model/translate.py +++ b/src/model/translate.py @@ -19,7 +19,7 @@ if 'Unnamed: 0' not in df.columns: exit(1) # Take a sample of 10 entries -sample_size = 10 +sample_size = 5000 df_sample = df.sample(n=sample_size, random_state=42) # Load the translation model