Muhammad Taqi Raza
commited on
Commit
·
be42567
1
Parent(s):
4bbb2d9
readme update: gpu support
Browse files- gradio_app.py +0 -2
gradio_app.py
CHANGED
|
@@ -5,7 +5,6 @@ from pathlib import Path
|
|
| 5 |
import gradio as gr
|
| 6 |
import numpy as np
|
| 7 |
import os
|
| 8 |
-
from spaces import GPU
|
| 9 |
# -----------------------------
|
| 10 |
# Setup paths and env
|
| 11 |
# -----------------------------
|
|
@@ -111,7 +110,6 @@ def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
|
|
| 111 |
# -----------------------------
|
| 112 |
# Step 2: Run Inference
|
| 113 |
# -----------------------------
|
| 114 |
-
@GPU
|
| 115 |
def inference(
|
| 116 |
fps, num_frames, controlnet_weights, controlnet_guidance_start,
|
| 117 |
controlnet_guidance_end, guidance_scale, num_inference_steps, dtype,
|
|
|
|
| 5 |
import gradio as gr
|
| 6 |
import numpy as np
|
| 7 |
import os
|
|
|
|
| 8 |
# -----------------------------
|
| 9 |
# Setup paths and env
|
| 10 |
# -----------------------------
|
|
|
|
| 110 |
# -----------------------------
|
| 111 |
# Step 2: Run Inference
|
| 112 |
# -----------------------------
|
|
|
|
| 113 |
def inference(
|
| 114 |
fps, num_frames, controlnet_weights, controlnet_guidance_start,
|
| 115 |
controlnet_guidance_end, guidance_scale, num_inference_steps, dtype,
|