init progress

app.py

@@ -57,14 +57,13 @@ def inference(x, count_only_api, api_key,
               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)
+    progress(0, desc="Starting...")
     # check if GPU is available
     if torch.cuda.is_available():
         providers = [("CUDAExecutionProvider", {"device_id": torch.cuda.current_device(),
                                                 "user_compute_stream": str(torch.cuda.current_stream().cuda_stream)})]
         sess_options = ort.SessionOptions()
         ort_sess = ort.InferenceSession(onnx_file, sess_options=sess_options, providers=providers)
-        print("Using GPU")
     else:
         ort_sess = ort.InferenceSession(onnx_file)
     #api = HfApi(token=os.environ['DATASET_SECRET'])
@@ -292,7 +291,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 rope speed', 'double dutch', 'double unders', 'single bounces', 'double bounces', 'triple unders', 'other'],
+        ticktext=['single\nrope', 'double\ndutch', 'double\nunders', 'single\nbounces', 'double\nbounces', 'triple\nunders', 'other'],
         title='event type'
     ))