daKhosa commited on
Commit
1efde5f
·
1 Parent(s): 3831078

Remove ZeroGPU duration and log WanGP startup

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -289,15 +289,13 @@ class _QueueWriter(io.TextIOBase):
289
  self._real.flush()
290
 
291
 
292
- @spaces.GPU(duration=45)
293
  def generate_video(image, prompt, resolution, steps, guidance_scale, seconds, seed):
294
  if image is None:
295
  raise gr.Error("Please upload an image.")
296
  if not prompt.strip():
297
  raise gr.Error("Please enter a prompt.")
298
 
299
- session = _ensure_session()
300
-
301
  frames = int(seconds * 25 // 8) * 8 + 1
302
  image_path = str(Path(image.strip()).resolve())
303
  out_dir = Path(tempfile.mkdtemp())
@@ -330,6 +328,9 @@ def generate_video(image, prompt, resolution, steps, guidance_scale, seconds, se
330
  sys.stdout = writer
331
  sys.stderr = writer # tqdm and progress bars write to stderr
332
  try:
 
 
 
333
  result_box[0] = session.run_task(task)
334
  except Exception as exc:
335
  error_box[0] = exc
 
289
  self._real.flush()
290
 
291
 
292
+ @spaces.GPU
293
  def generate_video(image, prompt, resolution, steps, guidance_scale, seconds, seed):
294
  if image is None:
295
  raise gr.Error("Please upload an image.")
296
  if not prompt.strip():
297
  raise gr.Error("Please enter a prompt.")
298
 
 
 
299
  frames = int(seconds * 25 // 8) * 8 + 1
300
  image_path = str(Path(image.strip()).resolve())
301
  out_dir = Path(tempfile.mkdtemp())
 
328
  sys.stdout = writer
329
  sys.stderr = writer # tqdm and progress bars write to stderr
330
  try:
331
+ print("[startup] Loading WanGP session...")
332
+ session = _ensure_session()
333
+ print("[startup] WanGP session ready. Starting generation...")
334
  result_box[0] = session.run_task(task)
335
  except Exception as exc:
336
  error_box[0] = exc