Upload app.py

app.py

@@ -25,7 +25,7 @@ model = get_peft_model(model, lora_config)
 model.gradient_checkpointing_enable()   
 
 @spaces.GPU(duration=120)
-def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
+def fine_tune_model(dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
     try:
         torch.cuda.empty_cache()
         def compute_metrics(eval_pred):