Spaces:
Running on Zero
Running on Zero
daKhosa commited on
Commit ·
1efde5f
1
Parent(s): 3831078
Remove ZeroGPU duration and log WanGP startup
Browse files
app.py
CHANGED
|
@@ -289,15 +289,13 @@ class _QueueWriter(io.TextIOBase):
|
|
| 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 |
-
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
|