Update app.py

app.py

@@ -76,7 +76,7 @@ num_epoch = st.number_input("Number of epochs:",
                             min_value=1,
                             max_value=1000,
                             step=1,
-                            value=10,
+                            value=1,
                             )
 
 tonic_name = st.selectbox(
@@ -90,9 +90,14 @@ if generate_image_button:
     input_file = f"./examples/{model_name}"
     if uploaded_file is not None:
         input_file = uploaded_file.name
+        with open(input_file, 'wb') as f:
+            f.write(uploaded_file.getvalue())
+    # print(uploaded_file.getvalue())
     with st.spinner(text=f"Generating, this may take some time..."):
         before, after = inference(input_file, num_epoch, tonic_name)
     st.markdown("Before:")
     st.audio(before)
     st.markdown("After:")
     st.audio(after)
+    if uploaded_file is not None:
+        os.remove(input_file)