cpu update for docker

app.py

@@ -1,9 +1,14 @@
+import multiprocessing
 import onnxruntime
 from torchvision import transforms
 import torch
 import torch.nn.functional as F
 import gradio as gr
-ort_sess = onnxruntime.InferenceSession("RFNet.onnx",use_external_data_format=True)
+
+sess_options = onnxruntime.SessionOptions()
+sess_options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
+sess_options.intra_op_num_threads = multiprocessing.cpu_count()
+ort_sess = onnxruntime.InferenceSession("RFNet.onnx", sess_options=sess_options)
 
 
 preprocess_img = transforms.Compose([

requirements.txt

@@ -1,5 +1,5 @@
-torch==1.12.1+cpu
-torchvision==0.13.1+cpu
+torch==1.12.0
+torchvision==0.13.0
 Pillow==9.2.0
 gradio==3.2.0
 onnxruntime==1.12.1