test higher resolution

app.py

@@ -27,6 +27,8 @@ from hls_download import download_clips
 
 plt.style.use('dark_background')
 
+IMG_SIZE = 256
+
 onnx_file = hf_hub_download(repo_id="dylanplummer/ropenet", filename="nextjump.onnx", repo_type="model", token=os.environ['DATASET_SECRET'])
 # model_xml = hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.xml", repo_type="model", token=os.environ['DATASET_SECRET'])
 # hf_hub_download(repo_id="dylanplummer/ropenet", filename="model.mapping", repo_type="model", token=os.environ['DATASET_SECRET'])
@@ -46,7 +48,7 @@ else:
     ort_sess = ort.InferenceSession(onnx_file)
 
 # warmup inference
-ort_sess.run(None, {'video': np.zeros((4, 64, 3, 224, 224), dtype=np.float32)})
+ort_sess.run(None, {'video': np.zeros((4, 64, 3, IMG_SIZE, IMG_SIZE), dtype=np.float32)})
 
 
 class SquarePad:
@@ -86,7 +88,7 @@ def run_inference(batch_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, 
+              img_size=IMG_SIZE, seq_len=64, stride_length=32, stride_pad=3, batch_size=4, 
               miss_threshold=0.8, marks_threshold=0.5, median_pred_filter=True, center_crop=True, both_feet=True, 
               api_call=False,
               progress=gr.Progress()):