Update app.py

app.py

@@ -5,10 +5,10 @@ import onnxruntime as rt
 import cv2
 import json
 import gradio as gr
-#from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download
 import onnxruntime as rt
 
-#modele = 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"))
 
@@ -48,7 +48,7 @@ def center_crop(img, out_height, out_width):
     return img
 
 
-sess = rt.InferenceSession('efficientnet-lite4-11.onnx')
+sess = rt.InferenceSession(modele)
 
 def inference(img):
   img = cv2.imread(img)