Update app.py

app.py

@@ -8,7 +8,7 @@ import tempfile
 import shutil
 import requests
 from pathlib import Path
-temp_dir = tempfile.TemporaryDirectory()
+temp_dir = os.path.expanduser("~/app")
 global ckpt_temp_file
 global audio_temp_file
 global config_temp_file
@@ -68,7 +68,7 @@ ckpt = st.file_uploader("Choose your CKPT", type= 'ckpt')
 if ckpt is not None:
       #TEMP FUNCTION
     ckpt_temp_file = os.path.join(temp_dir.name, "ckpt_temp_file.ckpt")
-    with open(ckpt_temp_file, "wb") as temp:
+    with tempfile.NamedTemporaryFile(mode="wb", suffix='.ckpt', delete=False, dir=temp_dir) as temp:
         bytes_data = ckpt.getvalue()
         temp.write(bytes_data)
         st.success("File saved to: {}".format(ckpt_temp_file))
@@ -87,7 +87,7 @@ config = st.file_uploader("Choose your config", type= 'yaml')
     # Check if user uploaded a config file
 if config is not None:
       #TEMP FUNCTION
-    with tempfile.NamedTemporaryFile(mode="wb", suffix='.yaml', delete=False) as temp:
+    with tempfile.NamedTemporaryFile(mode="wb", suffix='.yaml', delete=False, dir=temp_dir) as temp:
         # Get the file contents as bytes
         bytes_data = config.getvalue()
         # Write the bytes to the temporary file
@@ -105,7 +105,7 @@ audio = st.file_uploader("Choose your audio", type=["wav", "mp3"])
     # Check if user uploaded an audio file
 if audio is not None:
   #TEMP FUNCTION
-    with tempfile.NamedTemporaryFile(mode="wb", suffix='.wav', delete=False) as temp:
+    with tempfile.NamedTemporaryFile(mode="wb", suffix='.wav', delete=False, dir=temp_dir) as temp:
     # Get the file contents as bytes
         bytes_data = audio.getvalue()
     # Write the bytes to the temporary file