nasrin2023ripa commited on
Commit
e43a6d5
1 Parent(s): 7636ed1
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,10 +11,10 @@ with open("color_types_encoded50.json", "r") as fp:
11
  colors = list(encode_color_types.keys())
12
 
13
  # Set the providers parameter to use CPUExecutionProvider as fallback
14
- #providers = ['CPUExecutionProvider']
15
- #inf_session = rt.InferenceSession('rainbow-genre-cover-classifier-quantized.onnx', providers=providers)
16
 
17
- inf_session = rt.InferenceSession('rainbow-genre-cover-classifier-quantized.onnx')
18
  input_name = inf_session.get_inputs()[0].name
19
  output_name = inf_session.get_outputs()[0].name
20
 
 
11
  colors = list(encode_color_types.keys())
12
 
13
  # Set the providers parameter to use CPUExecutionProvider as fallback
14
+ providers = ['CPUExecutionProvider']
15
+ inf_session = rt.InferenceSession('rainbow-genre-cover-classifier-quantized.onnx', providers=providers)
16
 
17
+ #inf_session = rt.InferenceSession('rainbow-genre-cover-classifier-quantized.onnx')
18
  input_name = inf_session.get_inputs()[0].name
19
  output_name = inf_session.get_outputs()[0].name
20