progress bar

app.py

@@ -52,7 +52,11 @@ def sigmoid(x):
 
 
 @spaces.GPU()
-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):
+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)):
     print(x)
     # check if GPU is available
     if torch.cuda.is_available():
@@ -269,20 +273,21 @@ def inference(x, count_only_api, api_key, img_size=224, seq_len=64, stride_lengt
                     template="plotly_dark")
     
     fig.update_layout(legend=dict(
-        orientation="h",
-        yanchor="bottom",
-        y=0.98,
-        xanchor="right",
-        x=1,
-        font=dict(
-            family="Courier",
-            size=12,
-            color="black"
-            ),
-        bgcolor="AliceBlue",
-    ),
-    paper_bgcolor='rgba(0,0,0,0)',
-    plot_bgcolor='rgba(0,0,0,0)'
+            orientation="h",
+            yanchor="bottom",
+            y=0.98,
+            xanchor="right",
+            x=1,
+            font=dict(
+                family="Courier",
+                size=12,
+                color="black"
+                ),
+            bgcolor="AliceBlue",
+        ),
+        paper_bgcolor='rgba(0,0,0,0)',
+        plot_bgcolor='rgba(0,0,0,0)',
+        marker_line_width=0
     )
     fig.update_layout(coloraxis_colorbar=dict(
         tickvals=event_type_tick_vals,