fffiloni commited on
Commit
6081ae2
1 Parent(s): ea0c62f

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +3 -3
gradio_app.py CHANGED
@@ -83,7 +83,7 @@ def infer(frame1_path, frame2_path):
83
  num_inference_steps = 25
84
  noise_injection_steps = 0
85
  noise_injection_ratio = 0.5
86
- weighted_average = True
87
 
88
  generator = torch.Generator(device)
89
  if seed is not None:
@@ -91,10 +91,10 @@ def infer(frame1_path, frame2_path):
91
 
92
 
93
  frame1 = load_image(frame1_path)
94
- frame1 = frame1.resize((1024, 576))
95
 
96
  frame2 = load_image(frame2_path)
97
- frame2 = frame2.resize((1024, 576))
98
 
99
  frames = pipe(image1=frame1, image2=frame2,
100
  num_inference_steps=num_inference_steps, # 50
 
83
  num_inference_steps = 25
84
  noise_injection_steps = 0
85
  noise_injection_ratio = 0.5
86
+ weighted_average = False
87
 
88
  generator = torch.Generator(device)
89
  if seed is not None:
 
91
 
92
 
93
  frame1 = load_image(frame1_path)
94
+ frame1 = frame1.resize((512, 288))
95
 
96
  frame2 = load_image(frame2_path)
97
+ frame2 = frame2.resize((512, 288))
98
 
99
  frames = pipe(image1=frame1, image2=frame2,
100
  num_inference_steps=num_inference_steps, # 50