Updated app.py

app.py

@@ -32,8 +32,7 @@ def run_nnunet_predict(nifti_file):
 
     # Save the uploaded file to the input directory
     input_path = os.path.join(INPUT_DIR, "image.nii.gz")
-    with open(input_path, "wb") as f:
-        f.write(nifti_file.file.read())  # Use .file to access the underlying file object
+    os.rename(nifti_file.name, input_path)  # Move the uploaded file to the expected input location
 
     # Set environment variables for nnUNet
     os.environ["nnUNet_raw"] = DATASET_DIR
@@ -57,6 +56,7 @@ def run_nnunet_predict(nifti_file):
     except subprocess.CalledProcessError as e:
         return f"Error: {e}"
 
+
 # Gradio Interface
 interface = gr.Interface(
     fn=run_nnunet_predict,
@@ -71,4 +71,4 @@ download_model()
 
 # Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)