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 Repository
+from huggingface_hub import HfApi
 from transformers import AutoModelForCausalLM, AutoTokenizer, Seq2SeqTrainer, Seq2SeqTrainingArguments, pipeline
 
 ACTION_1 = "Prompt base model"
@@ -116,15 +116,14 @@ def fine_tune_model(base_model_name, dataset_name):
 
     # Save model to HF
 
-    repo = Repository(
-        local_dir="./output",
+    api = HfApi()
+    api.upload_file(
+        path_or_file="./output",
         repo_id=FT_MODEL_NAME,
         repo_type="model",
         use_auth_token=True,
     )
     
-    repo.push_to_hub()
-    
 def prompt_model(model_name, system_prompt, user_prompt, sql_context):
     pipe = pipeline("text-generation", 
                     model=model_name,