Luigi commited on
Commit
18eb8e3
·
1 Parent(s): 9f60e86

Fix Input Type Error raised in infer()

Browse files
Files changed (1) hide show
  1. convert_to_mixed.py +2 -2
convert_to_mixed.py CHANGED
@@ -34,8 +34,8 @@ def load_and_preprocess_image(image_path, size=(224, 224)):
34
 
35
  def infer(model_path, input_feed):
36
  session = ort.InferenceSession(model_path, providers=PROVIDERS)
37
- input_name = session.get_inputs()[0].name
38
- result = session.run(None, {input_name: input_feed})
39
  return result
40
 
41
  def main(args):
 
34
 
35
  def infer(model_path, input_feed):
36
  session = ort.InferenceSession(model_path, providers=PROVIDERS)
37
+ #input_name = session.get_inputs()[0].name
38
+ result = session.run(None, input_feed)
39
  return result
40
 
41
  def main(args):