fixed uploade name bug

app.py

@@ -352,7 +352,7 @@ if 'xq' not in st.session_state:
         prompt = start[2].text_input(
             "Prompt:", value="", placeholder="Examples: playing corgi, 女人举着雨伞, mouette volant au-dessus de la mer, ガラスの花瓶の花 ...")
         if len(prompt) > 0:
-            st.session_state.prompt = prompt
+            st.session_state.prompt = prompt.replace(' ', '_')
         start[3].markdown(
             '<p style="color:gray;"> Don\'t know what to search? Try <b>Random</b>!</p>\
             <p>🌟 We also support multi-language search. Type any language you know to search! ⌨️ </p>',
@@ -384,6 +384,7 @@ if 'xq' not in st.session_state:
             assert len(weights) == 1
             xq = numpy_helper.to_array(weights[0]).tolist()
             assert len(xq) == DIMS
+            st.session_state.prompt = upld_model.name.split(".onnx")[0].replace(' ', '_')
         else:
             print(f"Input prompt is {prompt}")
             # Tokenize the vectors
@@ -432,7 +433,7 @@ if 'xq' in st.session_state:
                     st.markdown(card_with_conf(i, dist, url),
                                 unsafe_allow_html=True)
         dnld_nam = st.text_input('Download Name:',
-                                 f'{(st.session_state.prompt if "prompt" in st.session_state else (upld_model.name.split(".onnx")[0] if upld_model is not None else "model"))}.onnx',
+                                 f'{(st.session_state.prompt if "prompt" in st.session_state else "model")}.onnx',
                                  max_chars=50)
         dnld_btn = st.download_button('Download your classifier!',
                                       st.session_state.dnld_model,