:hammer_and_pick: Add a custom commit message to detect model upload laptop or huggingface space

app.py

@@ -488,14 +488,18 @@ def start(classifier_model, model_save_path, is_attention_model=False, m_optimiz
   classifier_model.save_pretrained(model_subdirectory)
 
   # push to the hub
+  commit_message = f":tada: Push model for window size {WINDOW} from huggingface space"
+  if is_my_laptop:
+    commit_message = f":tada: Push model for window size {WINDOW} from zephyrus"
+
   classifier_model.push_to_hub(
     repo_id="fahimfarhan/mqtl-classifier-model",
     # subfolder=f"my-awesome-model-{WINDOW}", subfolder didn't work :/
-    commit_message=f":tada: Push model for window size {WINDOW}"
+    commit_message=commit_message # f":tada: Push model for window size {WINDOW}"
   )
 
   # reload
-  model = classifier_model.from_pretrained(f"my-awesome-model-{WINDOW}")
+  classifier_model = classifier_model.from_pretrained(f"my-awesome-model-{WINDOW}")
   # repo_url = "https://huggingface.co/fahimfarhan/mqtl-classifier-model"
   #
   # push_to_hub(
@@ -514,12 +518,12 @@ def start(classifier_model, model_save_path, is_attention_model=False, m_optimiz
 if __name__ == '__main__':
   login_inside_huggingface_virtualmachine()
 
-  WINDOW = 200
+  WINDOW = 4000
   simple_cnn = Cnn1dClassifier(seq_len=WINDOW)
   simple_cnn.enable_logging = True
 
   start(classifier_model=simple_cnn, model_save_path=simple_cnn.file_name, WINDOW=WINDOW,
-        dataset_folder_prefix="inputdata/", is_debug=True, max_epochs=10)
+        dataset_folder_prefix="inputdata/", is_debug=True, max_epochs=3)
 
   pass