fix: downgrade torch and torchvision version (#347)

gooishin · gooishin · commit eb2481698045 · 2025-02-04T14:17:46.000+08:00
diff --git a/usecases/ai/microservices/image-classification/README.md b/usecases/ai/microservices/image-classification/README.md
@@ -19,6 +19,8 @@ Package | Version
 --- | ---
 openvino-dev | 2024.6.0
 grpcio | 1.69.0
+torch | 2.5.1
+torchvision | 0.20.1
 tensorflow-serving-api | 2.17.1
 opencv-python | 4.10.0.84
 
diff --git a/usecases/ai/microservices/image-classification/requirements.txt b/usecases/ai/microservices/image-classification/requirements.txt
@@ -1,4 +1,6 @@
 openvino-dev[onnx,pytorch,tensorflow]==2024.6.0
+torch==2.5.1
+torchvision==0.20.1
 grpcio==1.69.0
 tensorflow-serving-api==2.17.1
 opencv-python==4.10.0.84
