Update app.py

app.py

@@ -423,10 +423,16 @@ def inference(in_video, stream_url, start_time, end_time, beep_detection_on, eve
         self_pct_err = 0
     total_confidence = confidence * (1 - self_pct_err)
 
-    if both_feet:
-        count_msg = f"## Reps Count (both feet): {count_pred:.1f}, Marks Count (both feet): {marks_count_pred:.1f}, Confidence: {total_confidence:.2f}"
+    if LOCAL:
+        if both_feet:
+            count_msg = f"## Reps Count (both feet): {count_pred:.1f}, Marks Count (both feet): {marks_count_pred:.1f}, Confidence: {total_confidence:.2f}"
+        else:
+            count_msg = f"## Reps Count (one foot): {count_pred:.1f}, Marks Count (one foot): {marks_count_pred:.1f}, Confidence: {total_confidence:.2f}"
     else:
-        count_msg = f"## Reps Count (one foot): {count_pred:.1f}, Marks Count (one foot): {marks_count_pred:.1f}, Confidence: {total_confidence:.2f}"
+        if both_feet:
+            count_msg = f"## Reps Count (both feet): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
+        else:
+            count_msg = f"## Reps Count (one foot): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
 
     if api_call:
         if CACHE_API_CALLS:
@@ -638,7 +644,7 @@ with gr.Blocks() as demo:
                            outputs=[period_length], api_name='inference')
     examples = [
         #['https://hiemdall-dev2.azurewebsites.net/api/clip/clp_vrpWTyjM/mp4', '00:00:00', '00:01:10', True, 60],
-        ['files/wc2023.mp4', '', '00:00:00', '', True, 30, False, '', '0.2'],
+        ['files/wc2023.mp4', '', '00:00:00', '', True, 30, False, '30', '0.2'],
         #['https://hiemdall-dev2.azurewebsites.net/api/playlist/rec_rd2FAyUo/vod', '01:24:22', '01:25:35', True, 60]
         #['https://hiemdall-dev2.azurewebsites.net/api/playlist/rec_PY5Ukaua/vod, '00:52:53', '00:55:00', True, 120]
     ]