test to put online the model

app.py

@@ -24,6 +24,8 @@ from streamlit_cropper import st_cropper
 from streamlit_drawable_canvas import st_canvas
 from streamlit_image_select import image_select
 
+from huggingface_hub import PyTorchModelHubMixin
+
 
 def get_memory_usage():
     process = psutil.Process()
@@ -177,6 +179,13 @@ def load_models():
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         model_arrow.load_state_dict(torch.load(output_arrow, map_location=device))
         model_object.load_state_dict(torch.load(output_object, map_location=device))
+
+        # save locally
+        model_arrow.save_pretrained("model_arrow")
+
+        # push to the hub
+        model_arrow.push_to_hub("model_arrow")
+
         st.session_state.model_loaded = True
         st.session_state.model_arrow = model_arrow
         st.session_state.model_object = model_object

modules/toXML.py

@@ -129,8 +129,6 @@ def create_bpmn_object(process, bpmnplane, text_mapping, definitions, size, data
     positions = data['boxes']
     links = data['links']
 
-    print(elements)
-
     for i in keep_elements:
         element_id = elements[i]