linoyts HF Staff commited on
Commit
176f1a8
·
verified ·
1 Parent(s): 3113790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -88,7 +88,15 @@ def handle_image_upload(image):
88
 
89
  return gr.update(value=best_size)
90
 
91
- def get_duration(image, prompt, size, duration_seconds, sampling_steps, guide_scale, shift, seed):
 
 
 
 
 
 
 
 
92
  """Calculate dynamic GPU duration based on parameters."""
93
  if sampling_steps > 35 and duration_seconds > 2:
94
  return 95
 
88
 
89
  return gr.update(value=best_size)
90
 
91
+ def get_duration(image,
92
+ prompt,
93
+ size,
94
+ duration_seconds,
95
+ sampling_steps,
96
+ guide_scale,
97
+ shift,
98
+ seed,
99
+ progress):
100
  """Calculate dynamic GPU duration based on parameters."""
101
  if sampling_steps > 35 and duration_seconds > 2:
102
  return 95