Update app.py
Browse files
app.py
CHANGED
|
@@ -82,6 +82,33 @@ def predict(image: Image.Image):
|
|
| 82 |
|
| 83 |
return sp_text, st_text
|
| 84 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 85 |
demo = gr.Interface(
|
| 86 |
fn=predict,
|
| 87 |
inputs=gr.Image(type="pil"),
|
|
@@ -90,7 +117,7 @@ demo = gr.Interface(
|
|
| 90 |
gr.Textbox(label="Predicted state"),
|
| 91 |
],
|
| 92 |
title="EfficientNet Two-Head Layer Trap Nest (LTN) Classifier",
|
| 93 |
-
description=
|
| 94 |
examples=[
|
| 95 |
["./0.jpg"],
|
| 96 |
["./1.jpg"],
|
|
@@ -116,31 +143,6 @@ demo = gr.Interface(
|
|
| 116 |
["./21.jpg"],
|
| 117 |
["./22.jpg"],
|
| 118 |
],
|
| 119 |
-
descriptions=[
|
| 120 |
-
"0: Cacoxnus indagator - Lv",
|
| 121 |
-
"1: Chelostoma florisomne - DauLv",
|
| 122 |
-
"2: Chelostoma florisomne - OldFood",
|
| 123 |
-
"3: Coeliopencyrtus - DauLv",
|
| 124 |
-
"4: Eumenidae - DauLv",
|
| 125 |
-
"5: Eumenidae - OldFood",
|
| 126 |
-
"6: Heriades - DeadLv",
|
| 127 |
-
"7: Heriades - Lv",
|
| 128 |
-
"8: Heriades - OldFood",
|
| 129 |
-
"9: Hylaeus - Lv",
|
| 130 |
-
"10: Hylaeus - OldFood",
|
| 131 |
-
"11: Megachile - DauLv",
|
| 132 |
-
"12: Osmia bicornis - DauLv",
|
| 133 |
-
"13: Osmia bicornis - DeadLv",
|
| 134 |
-
"14: Osmia bicornis - OldFood",
|
| 135 |
-
"15: Osmia cornuta - DauLv",
|
| 136 |
-
"16: Osmia cornuta - DeadLv",
|
| 137 |
-
"17: Osmia cornuta - OldFood",
|
| 138 |
-
"18: Passaloecus - Lv",
|
| 139 |
-
"19: Passaloecus - OldFood",
|
| 140 |
-
"20: Psenulus - DauLv",
|
| 141 |
-
"21: Trypoxylon - DauLv",
|
| 142 |
-
"22: Trypoxylon - OldFood",
|
| 143 |
-
],
|
| 144 |
)
|
| 145 |
|
| 146 |
if __name__ == "__main__":
|
|
|
|
| 82 |
|
| 83 |
return sp_text, st_text
|
| 84 |
|
| 85 |
+
EXAMPLE_TEXT = """
|
| 86 |
+
12 species, 4 states supported:
|
| 87 |
+
0: Cacoxnus indagator - Lv
|
| 88 |
+
1: Chelostoma florisomne - DauLv
|
| 89 |
+
2: Chelostoma florisomne - OldFood
|
| 90 |
+
3: Coeliopencyrtus - DauLv
|
| 91 |
+
4: Eumenidae - DauLv
|
| 92 |
+
5: Eumenidae - OldFood
|
| 93 |
+
6: Heriades - DeadLv
|
| 94 |
+
7: Heriades - Lv
|
| 95 |
+
8: Heriades - OldFood
|
| 96 |
+
9: Hylaeus - Lv
|
| 97 |
+
10: Hylaeus - OldFood
|
| 98 |
+
11: Megachile - DauLv
|
| 99 |
+
12: Osmia bicornis - DauLv
|
| 100 |
+
13: Osmia bicornis - DeadLv
|
| 101 |
+
14: Osmia bicornis - OldFood
|
| 102 |
+
15: Osmia cornuta - DauLv
|
| 103 |
+
16: Osmia cornuta - DeadLv
|
| 104 |
+
17: Osmia cornuta - OldFood
|
| 105 |
+
18: Passaloecus - Lv
|
| 106 |
+
19: Passaloecus - OldFood
|
| 107 |
+
20: Psenulus - DauLv
|
| 108 |
+
21: Trypoxylon - DauLv
|
| 109 |
+
22: Trypoxylon - OldFood
|
| 110 |
+
"""
|
| 111 |
+
|
| 112 |
demo = gr.Interface(
|
| 113 |
fn=predict,
|
| 114 |
inputs=gr.Image(type="pil"),
|
|
|
|
| 117 |
gr.Textbox(label="Predicted state"),
|
| 118 |
],
|
| 119 |
title="EfficientNet Two-Head Layer Trap Nest (LTN) Classifier",
|
| 120 |
+
description=EXAMPLE_TEXT,
|
| 121 |
examples=[
|
| 122 |
["./0.jpg"],
|
| 123 |
["./1.jpg"],
|
|
|
|
| 143 |
["./21.jpg"],
|
| 144 |
["./22.jpg"],
|
| 145 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 146 |
)
|
| 147 |
|
| 148 |
if __name__ == "__main__":
|