Yaron Koresh commited on
Commit
fcc5d4e
·
verified ·
1 Parent(s): 53b9153

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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