nasrin2023ripa
commited on
Commit
•
e43a6d5
1
Parent(s):
7636ed1
line 17
Browse files
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 |
-
|
15 |
-
|
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 |
|