Update requirements.txt

requirements.txt

@@ -17,7 +17,7 @@ torchgeometry
 async-lru==2.0.4
 
 # note: gradio is only used for the cropping utility
-gradio==5.39.0
+gradio==5.6.0
 
 pyyaml==6.0.1
 numpy==1.22.0 # Updated to resolve conflicts
@@ -36,7 +36,14 @@ ffmpeg-python==0.2.0
 
 # versions of onnx gpu <= 1.17 did not support Cuda 12
 onnxruntime-gpu==1.19.2
-onnx==1.16.2
+
+# onnx 1.16.2 has some issues it seems:
+# https://github.com/onnx/onnx/issues/6267
+#onnx==1.16.2
+# update: I've rolled it back to 1.16.1
+# see: https://github.com/jbilcke-hf/FacePoke/issues/23#issuecomment-2414714490
+onnx==1.16.1
+
 scikit-image==0.20.0
 albumentations==1.3.1
 matplotlib==3.7.2
@@ -73,4 +80,4 @@ optimum-quanto==0.2.4
 pillow==10.4.0
 
 # by popular demand, let's add AVIF
-pillow-avif-plugin==1.4.6
+pillow-avif-plugin==1.4.6