dev

app.py

@@ -12,9 +12,9 @@ from huggingface_hub import login
 @spaces.GPU
 def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
     try:
-        login(api_key.strip())
+        #login(api_key.strip())
         # Load the model and tokenizer
-        model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2, force_download=True)
+        model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2)
         
     
         # Set training arguments