Update finetune.py

finetune.py

@@ -89,12 +89,12 @@ def fine_tune(model_name, finetune_type, model, tokenizer, summarize_text, train
     max_output = 60
     train_dataset = train
     eval_dataset = val
-    if finetune_type == "multilingual":
-        train_dataset = Dataset.from_pandas(train.sample(1200))
-        eval_dataset = Dataset.from_pandas(val.sample(150))
-    else:
+    if finetune_type == "english":
         train_dataset = Dataset.from_pandas(train.sample(1500))
         eval_dataset = Dataset.from_pandas(val.sample(200))
+    else:
+        train_dataset = Dataset.from_pandas(train.sample(1200))
+        eval_dataset = Dataset.from_pandas(val.sample(150))
     def preprocess_function(examples):
         inputs = [f"Summarize the text: {ex}" for ex in examples["source"]]
         targets = [f"Summary: {ex}" for ex in examples["target"]]