Update app.py

app.py

@@ -151,7 +151,7 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180, center=
     plt.close()
     
 # Function to run nnUNet inference
-@spaces.GPU  # Decorate the function to allocate GPU for its execution
+@spaces.GPU(duration=70)  # Decorate the function to allocate GPU for its execution
 def run_nnunet_predict(nifti_file):
     # Prepare directories
     os.makedirs(INPUT_DIR, exist_ok=True)