Hugging Face's logo

Spaces:
tedlasai
/
learn2refocus
Running on Zero

App Files Community
tedlasai commited on
Commit
90a371d
·
1 Parent(s): 3729b71

fix

Browse files
Files changed (2) hide show
  1. __pycache__/simple_inference.cpython-310.pyc +0 -0
  2. app.py +1 -1
__pycache__/simple_inference.cpython-310.pyc CHANGED
Binary files a/__pycache__/simple_inference.cpython-310.pyc and b/__pycache__/simple_inference.cpython-310.pyc differ
 
app.py CHANGED
@@ -45,7 +45,7 @@ def generate_vstack_from_image(image: Image.Image, input_focal_position: int, nu
45
 
46
  pipe.to(args.device)
47
  batch = convert_to_batch(image, input_focal_position=input_focal_position)
48
- output_frames, focal_stack_num = inference_on_image(args, batch, pipeline, device)
49
  save_dir = os.path.join(OUTPUT_DIR, batch['name'])
50
 
51
  write_output(save_dir, output_frames, focal_stack_num, batch['icc_profile'])
 
45
 
46
  pipe.to(args.device)
47
  batch = convert_to_batch(image, input_focal_position=input_focal_position)
48
+ output_frames, focal_stack_num = inference_on_image(args, batch, pipe, device)
49
  save_dir = os.path.join(OUTPUT_DIR, batch['name'])
50
 
51
  write_output(save_dir, output_frames, focal_stack_num, batch['icc_profile'])