fix progress

app.py

@@ -56,7 +56,7 @@ def inference(x, count_only_api, api_key,
               img_size=224, seq_len=64, stride_length=32, stride_pad=3, batch_size=4, 
               miss_threshold=0.8, marks_threshold=0.6, median_pred_filter=True, center_crop=True, both_feet=True, 
               api_call=False,
-              progress=gr.Progress(track_tqdm=True)):
+              progress=gr.Progress()):
     progress(0, desc="Starting...")
     # check if GPU is available
     if torch.cuda.is_available():
@@ -156,6 +156,7 @@ def inference(x, count_only_api, api_key,
                 event_type_logit_overlaps[idx:idx+seq_len] += 1
             batch_list = []
             idx_list = []
+        progress(i / (length + stride_length - stride_pad), desc="Processing...")
     if len(batch_list) != 0:  # still some leftover frames
         while len(batch_list) != batch_size:
             batch_list.append(batch_list[-1])
@@ -291,7 +292,7 @@ def inference(x, count_only_api, api_key,
     fig.update_traces(marker_line_width = 0)
     fig.update_layout(coloraxis_colorbar=dict(
         tickvals=event_type_tick_vals,
-        ticktext=['single\nrope', 'double\ndutch', 'double\nunders', 'single\nbounces', 'double\nbounces', 'triple\nunders', 'other'],
+        ticktext=['single<br>rope', 'double<br>dutch', 'double<br>unders', 'single<br>bounces', 'double<br>bounces', 'triple<br>unders', 'other'],
         title='event type'
     ))
 
@@ -304,7 +305,7 @@ def inference(x, count_only_api, api_key,
     
     # make a bar plot of the event type distribution
 
-    bar = px.bar(x=['single rope speed', 'double dutch', 'double unders', 'single bounces', 'double bounces', 'triple unders', 'other'], 
+    bar = px.bar(x=['single rope', 'double dutch', 'double unders', 'single bounces', 'double bounces', 'triple unders', 'other'], 
                  y=event_type_probs,
                  template="plotly_dark",
                  title="Event Type Distribution",