Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os, torch
 from datasets import load_dataset
-from huggingface_hub import HfApi, login
+from huggingface_hub import push_to_hub
 from transformers import AutoModelForCausalLM, AutoTokenizer, Seq2SeqTrainer, Seq2SeqTrainingArguments, pipeline
 
 ACTION_1 = "Prompt base model"
@@ -106,10 +106,15 @@ def fine_tune_model(base_model_name, dataset_name):
     )
 
     # Train model
-    trainer.train()
+    #trainer.train()
 
     # Save model to HF
-    trainer.push_to_hub(FT_MODEL_NAME, use_auth_token=True)
+    push_to_hub(
+        local_dir="./output",
+        repo_id=FT_MODEL_NAME,
+        repo_type="model",
+        use_auth_token=True,
+    )
     
 def prompt_model(model_name, system_prompt, user_prompt, sql_context):
     pipe = pipeline("text-generation",