bstraehle commited on
Commit
5eedbaa
1 Parent(s): e97ba92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -117,11 +117,11 @@ def fine_tune_model(base_model_name, dataset_name):
117
  # Save model to HF
118
 
119
  api = HfApi()
120
- api.upload_file(
121
- file_path="./output",
 
122
  repo_id=FT_MODEL_NAME,
123
  repo_type="model",
124
- use_auth_token=True,
125
  )
126
 
127
  def prompt_model(model_name, system_prompt, user_prompt, sql_context):
 
117
  # Save model to HF
118
 
119
  api = HfApi()
120
+
121
+ api.upload_folder(
122
+ folder_path="./output",
123
  repo_id=FT_MODEL_NAME,
124
  repo_type="model",
 
125
  )
126
 
127
  def prompt_model(model_name, system_prompt, user_prompt, sql_context):