Spaces:

smhavens
/

AnalogyArcade

Sleeping

smhavens commited on Dec 7, 2023

Commit

6ffe1b1

1 Parent(s): 0114c21

update save_to_hub

Files changed (1) hide show

app.py CHANGED Viewed

@@ -162,7 +162,7 @@ def finetune(train_dataloader):
     model.save_to_hub(
-        "sentence-transformers/all-MiniLM-L6-v2",
         organization="smhavens",
         train_datasets=["ag_news"],
         )

     model.save_to_hub(
+        "all-MiniLM-agNews",
         organization="smhavens",
         train_datasets=["ag_news"],
         )