add fix and push ft

app.py

@@ -47,7 +47,7 @@ def get_ir_evaluator(eval_ds):
 
 
 @spaces.GPU(duration=3600)
-def train(hf_token, dataset_id, model_id, num_epochs, dev):
+def train(hf_token, dataset_id, model_id, num_epochs, dev=True):
 
     ds = load_dataset(dataset_id, split="train", token=hf_token)
     ds = ds.shuffle(seed=42)
@@ -110,6 +110,8 @@ def train(hf_token, dataset_id, model_id, num_epochs, dev):
     print(ir_evaluator.primary_metric)
     print(ft_metrics[ir_evaluator.primary_metric])
 
+    if not dev: model.push_to_hub("fine-tuned-sentence-transformer", private=True, token=hf_token)
+
 
     metrics = pd.DataFrame([base_metrics, ft_metrics]).T
     print(metrics)
@@ -119,5 +121,5 @@ def train(hf_token, dataset_id, model_id, num_epochs, dev):
 ## logs to UI
 # https://github.com/gradio-app/gradio/issues/2362#issuecomment-1424446778
 
-demo = gr.Interface(fn=greet, inputs=["text", "text", "text", "number", "bool"], outputs=["text"]) # "dataframe"
+demo = gr.Interface(fn=train, inputs=["text", "text", "text", "number", "bool"], outputs=["text"]) # "dataframe"
 demo.launch()