javad-rezaie commited on
Commit
5c59f34
1 Parent(s): c79ac10

updated to include probability.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"),