Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -81,6 +81,8 @@ def Piper(name,positive_prompt,motion):
|
|
81 |
global time
|
82 |
global last_motion
|
83 |
global progress
|
|
|
|
|
84 |
|
85 |
pipe.unet.load_state_dict(torch.load(hf_hub_download(base, "unet/diffusion_pytorch_model.bin"), map_location=device), strict=False)
|
86 |
progress((0, step))
|
@@ -192,6 +194,7 @@ def main():
|
|
192 |
global fps
|
193 |
global time
|
194 |
global last_motion
|
|
|
195 |
|
196 |
last_motion=None
|
197 |
fps=10
|
|
|
81 |
global time
|
82 |
global last_motion
|
83 |
global progress
|
84 |
+
global base
|
85 |
+
global device
|
86 |
|
87 |
pipe.unet.load_state_dict(torch.load(hf_hub_download(base, "unet/diffusion_pytorch_model.bin"), map_location=device), strict=False)
|
88 |
progress((0, step))
|
|
|
194 |
global fps
|
195 |
global time
|
196 |
global last_motion
|
197 |
+
global base
|
198 |
|
199 |
last_motion=None
|
200 |
fps=10
|