Update app.py
Browse files
app.py
CHANGED
@@ -23,20 +23,36 @@ def classify_image(image):
|
|
23 |
confidence = np.max(prediction[0])
|
24 |
|
25 |
if confidence < 0.6:
|
26 |
-
|
27 |
-
|
28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
|
30 |
|
31 |
iface = gr.Interface(
|
32 |
fn=classify_image,
|
33 |
inputs="image",
|
34 |
-
outputs=[
|
|
|
|
|
|
|
|
|
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()
|