fix

app.py

@@ -23,7 +23,7 @@ pipe, device = load_model(args)
 OUTPUT_DIR = Path("/tmp/output_stacks")
 OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
 
-
+@spaces.GPU(timeout=300, duration=80)
 def generate_vstack_from_image(image: Image.Image, input_focal_position: int, num_inference_steps: int) -> str:
     """
     Wrapper for Gradio. Takes an image and returns a video path.
@@ -46,7 +46,9 @@ def generate_vstack_from_image(image: Image.Image, input_focal_position: int, nu
     pipe.to(args.device)
     batch = convert_to_batch(args.image_path, input_focal_position=input_focal_position)
     output_frames, focal_stack_num = inference_on_image(args, batch, pipeline, device)
-    write_output(OUTPUT_DIR, output_frames, focal_stack_num, batch['icc_profile'])
+    save_dir = os.path.join(OUTPUT_DIR, batch['name'])
+
+    write_output(save_dir, output_frames, focal_stack_num, batch['icc_profile'])
 
     if not os.path.exists(output_path):
         raise gr.Error("Video generation failed: output file not found.")
@@ -65,7 +67,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
         - 🌐 **Project page:** <https://learn2refocus.github.io/>  
         - 💻 **Code:** <https://github.com/tedlasai/learn2refocus/>  
 
-        Upload an image specify the input focal position. Near - 5cm, Far - Infinity. Then, click "Generate stack" to generate a focal stack.
+        Upload an image specify the input focal position (these values correspond to iPhone API positions, but approximately linear in diopters (inverse meters): 0 - 5cm, 8 - Infinity). Then, click "Generate stack" to generate a focal stack.
         """
     )