Upload folder using huggingface_hub

.idea/workspace.xml

@@ -74,7 +74,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1706519257550</updated>
-      <workItem from="1706519262472" duration="4216000" />
+      <workItem from="1706519262472" duration="4411000" />
     </task>
     <servers />
   </component>

app.py

@@ -21,16 +21,17 @@ def tensor2im(var):
 
 def process(input_image, step=None):
     progress = gr.Progress()
+    progress(0, desc="Starting...")
     img = tfs(input_image)[None, ...].to(device)
+    b, *_ = img.shape
     if step == None:
         timesteps = 200
     else:
         timesteps = step
     with torch.no_grad():
         model.eval()
-        b, *_ = img.shape
         y_t = torch.randn_like(img)
-        for i, _ in progress.tqdm(reversed(range(0, timesteps)), desc='sampling loop time step', total=timesteps):
+        for i in progress.tqdm(reversed(range(0, timesteps)), desc='sampling loop time step', total=timesteps):
             t = torch.full((b,), i, device=img.device, dtype=torch.long)
             y_t = model(img, y_t, t)
             # out, _ = model.netG.restoration(img, sample_num=8)