Update app.py

app.py

@@ -214,13 +214,18 @@ def upload_to_huggingface(file_path, repo_id, token):
             repo_id=repo_id,
         )
         st.success(f"✅ File uploaded to Hugging Face: {repo_id}")
+
+        # Reset session state and rerun
+        st.session_state.quantized_model_path = None
+        st.session_state.upload_to_hf = False
+        st.rerun()
     except Exception as e:
         st.error(f"❌ Failed to upload file: {e}")
 
 st.title("🦙 LLaMA Model Quantization (llama.cpp)")
 
 
-selected_model = st.selectbox("Select the Hugging Face Model", models_list)
+selected_model = st.selectbox("Select the Hugging Face Model", models_list, index=None)
 hf_model_name = selected_model if selected_model else st.text_input("Enter Hugging Face Model (If not there in the above list)")
 f_map_type = st.selectbox("Select the GGUF conversion type", F_TYPE_MAP)
 
@@ -236,7 +241,10 @@ if start_button:
 
 if st.session_state.quantized_model_path:
     with open(st.session_state.quantized_model_path, "rb") as f:
-        st.download_button("⬇️ Download Quantized Model", f, file_name=os.path.basename(st.session_state.quantized_model_path))
+        if st.download_button("⬇️ Download Quantized Model", f, file_name=os.path.basename(st.session_state.quantized_model_path)):
+            st.session_state.quantized_model_path = None
+            st.session_state.upload_to_hf = False
+            st.rerun()
     
     # Checkbox for upload section
     st.session_state.upload_to_hf = st.checkbox("Upload to Hugging Face", value=st.session_state.upload_to_hf)