Nadine Rueegg commited on
Commit
bc8bb8d
·
1 Parent(s): 36c86a9

remove tqdm

Browse files
Files changed (1) hide show
  1. scripts/gradio_demo.py +1 -1
scripts/gradio_demo.py CHANGED
@@ -511,7 +511,7 @@ def run_bite_inference(input_image, bbox=None, apply_ttopt=True):
511
  total_loss.backward(retain_graph=True) #
512
  current_optimizer.step()
513
  current_scheduler.step(total_loss)
514
- loop.set_description(f"Body Fitting = {total_loss.item():.3f}")
515
 
516
  # save the result three times (0, 150, 300)
517
  if i == 300: # if i % 150 == 0:
 
511
  total_loss.backward(retain_graph=True) #
512
  current_optimizer.step()
513
  current_scheduler.step(total_loss)
514
+ # loop.set_description(f"Body Fitting = {total_loss.item():.3f}")
515
 
516
  # save the result three times (0, 150, 300)
517
  if i == 300: # if i % 150 == 0: