Kevin Fink commited on
Commit
d202762
·
1 Parent(s): f916b82
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,9 +12,9 @@ from huggingface_hub import login
12
  @spaces.GPU
13
  def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
14
  try:
15
- login(api_key.strip())
16
  # Load the model and tokenizer
17
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2, force_download=True)
18
 
19
 
20
  # Set training arguments
 
12
  @spaces.GPU
13
  def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
14
  try:
15
+ #login(api_key.strip())
16
  # Load the model and tokenizer
17
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2)
18
 
19
 
20
  # Set training arguments