F-G Fernandez commited on
Commit
bcbd897
1 Parent(s): fc36b00

fix: Fixed typo

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,10 +16,10 @@ from PIL import Image
16
  REPO = "frgfm/rexnet1_0x"
17
 
18
  # Download model config & checkpoint
19
- with open(hf_hub_download(args.repo, filename="config.json"), "rb") as f:
20
  cfg = json.load(f)
21
 
22
- ort_session = onnxruntime.InferenceSession(hf_hub_download(args.repo, filename="model.onnx"))
23
 
24
  def preprocess_image(pil_img: Image.Image) -> np.ndarray:
25
  """Preprocess an image for inference
16
  REPO = "frgfm/rexnet1_0x"
17
 
18
  # Download model config & checkpoint
19
+ with open(hf_hub_download(REPO, filename="config.json"), "rb") as f:
20
  cfg = json.load(f)
21
 
22
+ ort_session = onnxruntime.InferenceSession(hf_hub_download(REPO, filename="model.onnx"))
23
 
24
  def preprocess_image(pil_img: Image.Image) -> np.ndarray:
25
  """Preprocess an image for inference