Fix bug with deleting state

create.py

@@ -16,7 +16,8 @@ def _add_back_button():
 
 def _add_delete_button():
     def fn():
-        st.session_state.screen.state = "start"
+        if "hf_path" in st.session_state:
+            del st.session_state["hf_path"]
         if "model_card" in st.session_state:
             del st.session_state["model_card"]
         if "task_state" in st.session_state:
@@ -25,6 +26,7 @@ def _add_delete_button():
             del st.session_state["create_repo_name"]
         if "hf_token" in st.session_state:
             del st.session_state["hf_token"]
+        st.session_state.screen.state = "start"
 
     st.button("Delete", on_click=fn, help="Start over from scratch (lose all progress)")
 

edit.py

@@ -278,11 +278,13 @@ def reset_model_card() -> None:
 
 
 def delete_model_card() -> None:
-    st.session_state.screen.state = "start"
+    if "hf_path" in st.session_state:
+        del st.session_state["hf_path"]
     if "model_card" in st.session_state:
         del st.session_state["model_card"]
     if "task_state" in st.session_state:
         st.session_state.task_state.reset()
+    st.session_state.screen.state = "start"
 
 
 def undo_last():