dev

app.py

@@ -146,7 +146,7 @@ def predict(text):
 @spaces.GPU(duration=120)
 def run_train(dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
     config = AutoConfig.from_pretrained("google/t5-efficient-tiny")
-    model = AutoModelForSeq2SeqLM.from_config(config, num_labels=2)
+    model = AutoModelForSeq2SeqLM.from_config(config)
     result = fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad)
     return result
 # Create Gradio interface