DHEIVER commited on
Commit
077f973
1 Parent(s): c8c8206

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -6
app.py CHANGED
@@ -23,20 +23,36 @@ def classify_image(image):
23
  confidence = np.max(prediction[0])
24
 
25
  if confidence < 0.6:
26
- return "Unable to detect", confidence
27
-
28
- return class_names[predicted_class.numpy()], confidence
 
 
 
 
 
 
 
 
 
 
 
29
 
30
 
31
  iface = gr.Interface(
32
  fn=classify_image,
33
  inputs="image",
34
- outputs=["text", "number"],
 
 
 
 
35
  examples=[
36
  ['examples/0.jpg'],
37
  ['examples/1.jpg'],
38
  ['examples/2.jpg'],
39
- ['examples/3.jpg'],
 
 
40
 
41
- ])
42
  iface.launch()
 
23
  confidence = np.max(prediction[0])
24
 
25
  if confidence < 0.6:
26
+ report = {
27
+ 'result': "Unable to detect",
28
+ 'confidence': confidence,
29
+ 'details': "The model was unable to confidently classify the image."
30
+ }
31
+ else:
32
+ class_name = class_names[predicted_class.numpy()]
33
+ report = {
34
+ 'result': class_name,
35
+ 'confidence': confidence,
36
+ 'details': f"The image is classified as {class_name} with a confidence of {confidence:.2f}."
37
+ }
38
+
39
+ return report
40
 
41
 
42
  iface = gr.Interface(
43
  fn=classify_image,
44
  inputs="image",
45
+ outputs=[
46
+ gr.outputs.Textbox(label="Result"),
47
+ gr.outputs.Number(label="Confidence"),
48
+ gr.outputs.Textbox(label="Details")
49
+ ],
50
  examples=[
51
  ['examples/0.jpg'],
52
  ['examples/1.jpg'],
53
  ['examples/2.jpg'],
54
+ ['examples/3.jpg']
55
+ ]
56
+ )
57
 
 
58
  iface.launch()