Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,14 +50,14 @@ def get_server_status():
|
|
| 50 |
return {"cpu": f"{cpu_percent}%", "memory": f"{memory.percent}%", "disk": f"{disk.percent}%", "gpu": gpu_info}
|
| 51 |
|
| 52 |
|
| 53 |
-
@spaces.GPU(
|
| 54 |
def generate_vanilla(model_name, prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
|
| 55 |
engine = load_model(model_name)
|
| 56 |
video_path = generate(engine, prompt, num_inference_steps, guidance_scale)
|
| 57 |
return video_path
|
| 58 |
|
| 59 |
|
| 60 |
-
@spaces.GPU(
|
| 61 |
def generate_vs(
|
| 62 |
model_name,
|
| 63 |
prompt,
|
|
|
|
| 50 |
return {"cpu": f"{cpu_percent}%", "memory": f"{memory.percent}%", "disk": f"{disk.percent}%", "gpu": gpu_info}
|
| 51 |
|
| 52 |
|
| 53 |
+
@spaces.GPU()
|
| 54 |
def generate_vanilla(model_name, prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
|
| 55 |
engine = load_model(model_name)
|
| 56 |
video_path = generate(engine, prompt, num_inference_steps, guidance_scale)
|
| 57 |
return video_path
|
| 58 |
|
| 59 |
|
| 60 |
+
@spaces.GPU()
|
| 61 |
def generate_vs(
|
| 62 |
model_name,
|
| 63 |
prompt,
|