upgrade

README.md

@@ -4,7 +4,7 @@ emoji: 🦘
 colorFrom: blue
 colorTo: green
 sdk: gradio
-sdk_version: 3.44.3
+sdk_version: 4.13.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -243,7 +243,8 @@ def inference(x, count_only_api, api_key, img_size=192, seq_len=64, stride_lengt
                                 'jumps_size': (full_marks + 0.05) * 10,
                                 'miss_size': np.clip((1 - periodicity) * 0.9 + 0.1, 1, 10),
                                 'seconds': np.linspace(0, seconds, num=len(periodLength))})
-    
+    event_type_tick_vals = np.linspace(0, 1, num=7)
+    event_type_colors = ['red', 'orange', 'green', 'blue', 'purple', 'pink', 'black']
     fig = px.scatter(data_frame=df,
                     x='seconds', 
                     y='jumps per second',
@@ -252,7 +253,7 @@ def inference(x, count_only_api, api_key, img_size=192, seq_len=64, stride_lengt
                     color='event_type',
                     size='jumps_size',
                     size_max=10,
-                    color_continuous_scale='Rainbow',
+                    color_continuous_scale=[(t, c) for t, c in zip(event_type_tick_vals, event_type_colors)],
                     range_color=(0,1),
                     title="Jumping speed (jumps-per-second)",
                     trendline='rolling',
@@ -278,7 +279,7 @@ def inference(x, count_only_api, api_key, img_size=192, seq_len=64, stride_lengt
     plot_bgcolor='rgba(0,0,0,0)'
     )
     fig.update_layout(coloraxis_colorbar=dict(
-        tickvals=np.linspace(0, 1, num=7),
+        tickvals=event_type_tick_vals,
         ticktext=['single rope speed', 'double dutch', 'double unders', 'single bounces', 'double bounces', 'triple unders', 'other'],
         title='event type'
     ))