Spaces:
Sleeping
Sleeping
javad-rezaie
commited on
Commit
•
5c59f34
1
Parent(s):
c79ac10
updated to include probability.
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def pred_class(image_path):
|
|
38 |
result_infer = compiled_model([input_image])[output_layer]
|
39 |
result_index = np.argmax(result_infer)
|
40 |
|
41 |
-
return CAR_CLASSES[result_index] + "With probabilty of " + str(max(result_infer))
|
42 |
|
43 |
inputs_image = [
|
44 |
gr.components.Image(type="filepath", label="Upload A Car Image"),
|
|
|
38 |
result_infer = compiled_model([input_image])[output_layer]
|
39 |
result_index = np.argmax(result_infer)
|
40 |
|
41 |
+
return CAR_CLASSES[result_index] + " With probabilty of " + str(np.max(result_infer))
|
42 |
|
43 |
inputs_image = [
|
44 |
gr.components.Image(type="filepath", label="Upload A Car Image"),
|