init

app.py

@@ -17,10 +17,10 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
             bias="none"  # Bias handling
         )        
         # Load the dataset
-        dataset = load_dataset(dataset_name)
+        dataset = load_dataset(dataset_name.strip())
     
         # Load the model and tokenizer
-        model = AutoModelForSeq2SeqLM.from_pretrained(model_name, num_labels=2)
+        model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2)
         #model = get_peft_model(model, lora_config)
         tokenizer = AutoTokenizer.from_pretrained(model_name)
     
@@ -48,7 +48,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
             logging_dir='./logs',
             logging_steps=10,
             push_to_hub=True,
-            hub_model_id=hub_id,
+            hub_model_id=hub_id.strip(),
             fp16=True,
             #lr_scheduler_type='cosine',
         )