audrey06100 commited on
Commit
e9151ee
·
1 Parent(s): 9126ac5
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -863,6 +863,9 @@ with gr.Blocks() as demo:
863
 
864
  break_flag = False # indicate if the process has been interrupted by the user
865
  for i in range(stage2_info["totalBatchNum"]):
 
 
 
866
  # establish a temp folder
867
  try:
868
  os.mkdir(filepath+"temp_data/")
@@ -871,10 +874,6 @@ with gr.Blocks() as demo:
871
  break_flag = True
872
  break
873
 
874
- # update the running status
875
- md = "Running model({}/{})...".format(i+1, stage2_info["totalBatchNum"])
876
- yield {batch_md : gr.Markdown(md)}
877
-
878
  # get the mapped index order and the filled status for each tpl_channels
879
  new_idx = stage1_info["mappingData"][i]["newOrder"]
880
  fill_flags = stage1_info["mappingData"][i]["fillFlags"]
 
863
 
864
  break_flag = False # indicate if the process has been interrupted by the user
865
  for i in range(stage2_info["totalBatchNum"]):
866
+ md = "Running model({}/{})...".format(i+1, stage2_info["totalBatchNum"])
867
+ yield {batch_md : gr.Markdown(md)}
868
+
869
  # establish a temp folder
870
  try:
871
  os.mkdir(filepath+"temp_data/")
 
874
  break_flag = True
875
  break
876
 
 
 
 
 
877
  # get the mapped index order and the filled status for each tpl_channels
878
  new_idx = stage1_info["mappingData"][i]["newOrder"]
879
  fill_flags = stage1_info["mappingData"][i]["fillFlags"]