Update app.py
Browse files
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 |
-
|
121 |
-
|
|
|
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):
|