Update app.py

app.py

@@ -10,8 +10,11 @@ import wget
 import time
 
 def imageInput(device, src):
-    os.makedirs('data/uploads')
-    os.makedirs('data/outputs')
+    current_dir = os.getcwd()
+    upload_path = os.path.join(current_dir, 'data/uploads')
+    output_path = os.path.join(current_dir, 'data/outputs')
+    os.makedirs(upload_path)
+    os.makedirs(output_path)
     if src == 'Upload your own data.':
         image_file = st.file_uploader("Upload An Image", type=['png', 'jpeg', 'jpg'])
         col1, col2 = st.columns(2)