Update app.py

app.py

@@ -17,9 +17,9 @@ APP_DESCRIPTION = config['APP_DESCRIPTION']
 def contains_only_ascii(input_string):
     return all(ord(char) < 128 for char in input_string)
 
-def create_temp_file(input_mp3):
+def create_temp_file(input_wav):
     temp_file = tempfile.NamedTemporaryFile(delete=False)
-    temp_file.write(input_mp3.read())
+    temp_file.write(input_wav.read())
     return temp_file
 
 def remove_temp_file(temp_file):
@@ -38,7 +38,7 @@ st.markdown(APP_DESCRIPTION)
 input_wav = st.file_uploader("Upload a WAV file with your voice", type=["mp3"])
 clone_wav = st.file_uploader("Upload a WAV file with voice to clone", type=["mp3"])
 
-if input_mp3 and clone_mp3:
+if input_mp3 and clone_wav:
     progress_bar = st.progress(0)
     status_text = st.empty()
 
@@ -69,4 +69,4 @@ if input_mp3 and clone_mp3:
 
     st.audio(audio_bytes, format='audio/wav')
 
-    st.download_button('Download mp3', data=audio_bytes, file_name='output.wav')
+    st.download_button('Download wav', data=audio_bytes, file_name='output.wav')