restore

app.py

@@ -24,8 +24,6 @@ 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()
@@ -180,12 +178,6 @@ def load_models():
         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