dev

app.py

@@ -7,14 +7,14 @@ import traceback
 
 
 import os
-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())
         # Load the model and tokenizer
-        model = AutoModelForSeq2SeqLM.from_pretrained(model_name.strip(), num_labels=2)
+        model = AutoModelForSeq2SeqLM.from_pretrained('google/t5-efficient-tiny-nh8', num_labels=2)
         
     
         # Set training arguments
@@ -45,7 +45,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
         max_length = 128
         # Load the dataset
         dataset = load_dataset(dataset_name.strip())
-        tokenizer = AutoTokenizer.from_pretrained(model_name)
+        tokenizer = AutoTokenizer.from_pretrained('google/t5-efficient-tiny-nh8')
         # Tokenize the dataset
         def tokenize_function(examples):