Update app.py

app.py

@@ -5,11 +5,13 @@ 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
+import os
 
-modele = hf_hub_download(repo_id="onnx/EfficientNet-Lite4", filename="efficientnet-lite4-11.onnx")
-print(modele)
+os.system("wget https://github.com/onnx/models/raw/main/vision/classification/efficientnet-lite4/model/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 +50,7 @@ def center_crop(img, out_height, out_width):
     img = img[top:bottom, left:right]
     return img
 
-sess = rt.InferenceSession(modele)
+sess = rt.InferenceSession('efficientnet-lite4-11.onnx')
 
 def inference(img):
   img = cv2.imread(img)