Update app.py

app.py

@@ -6,8 +6,9 @@ import cv2
 import json
 import gradio as gr
 from huggingface_hub import hf_hub_download
+import onnxruntime as rt
 
-hf_hub_download(repo_id="onnx/EfficientNet-Lite4", filename="efficientnet-lite4-11.onnx")
+modele = hf_hub_download(repo_id="onnx/EfficientNet-Lite4", filename="efficientnet-lite4-11.onnx")
 # load the labels text file
 labels = json.load(open("labels_map.txt", "r"))
 
@@ -46,12 +47,7 @@ def center_crop(img, out_height, out_width):
     img = img[top:bottom, left:right]
     return img
 
-
-
-import onnxruntime as rt
-
-
-sess = rt.InferenceSession('efficientnet-lite4-11.onnx')
+sess = rt.InferenceSession(modele)
 
 def inference(img):
   img = cv2.imread(img)