Kevin Fink
commited on
Commit
·
d202762
1
Parent(s):
f916b82
dev
Browse files
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
|
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
|