tanvir-ishraq
commited on
Commit
•
b4a7d51
1
Parent(s):
c860493
Update app.py
Browse files
app.py
CHANGED
@@ -14,18 +14,21 @@ cap_labels = (
|
|
14 |
'Bayraktar TB2 UVAC drone',
|
15 |
'CH-5 Rainbow UVAC drone',
|
16 |
'G6 Rhino artillery',
|
17 |
-
'Hermes 900 drone',
|
|
|
18 |
'Humvee vehicle',
|
19 |
'LAV-25 vehicle',
|
20 |
'Leopard 2 tank',
|
21 |
'M1 Abrams tank',
|
22 |
'M109 artillery',
|
23 |
-
'M113 vehicle',
|
|
|
24 |
'MQ-9 Reaper UVAC drone',
|
25 |
'MRAP vehicle',
|
26 |
'RQ-4 Global Hawk UVAC drone',
|
27 |
'T-72 tank',
|
28 |
-
'Type 99 tank',
|
|
|
29 |
)
|
30 |
|
31 |
model = load_learner('models/ARMOR-classifier-v4.pkl')
|
@@ -48,5 +51,12 @@ examples = [
|
|
48 |
]
|
49 |
|
50 |
#interface with i/o
|
51 |
-
iface = gr.Interface(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
52 |
iface.launch(inline=False) # share=True for colab
|
|
|
14 |
'Bayraktar TB2 UVAC drone',
|
15 |
'CH-5 Rainbow UVAC drone',
|
16 |
'G6 Rhino artillery',
|
17 |
+
'Hermes 900 drone',
|
18 |
+
'Heron TP drone',
|
19 |
'Humvee vehicle',
|
20 |
'LAV-25 vehicle',
|
21 |
'Leopard 2 tank',
|
22 |
'M1 Abrams tank',
|
23 |
'M109 artillery',
|
24 |
+
'M113 vehicle',
|
25 |
+
'M270 MLRS artillery',
|
26 |
'MQ-9 Reaper UVAC drone',
|
27 |
'MRAP vehicle',
|
28 |
'RQ-4 Global Hawk UVAC drone',
|
29 |
'T-72 tank',
|
30 |
+
'Type 99 tank',
|
31 |
+
'smerch artillery'
|
32 |
)
|
33 |
|
34 |
model = load_learner('models/ARMOR-classifier-v4.pkl')
|
|
|
51 |
]
|
52 |
|
53 |
#interface with i/o
|
54 |
+
iface = gr.Interface(
|
55 |
+
fn=recognize_image,
|
56 |
+
inputs=image,
|
57 |
+
outputs=label,
|
58 |
+
examples=examples,
|
59 |
+
title="A.R.M.O.R - Armament Models Recognizer",
|
60 |
+
description="A comprehensive security measure image classification model that classifies (for now) 22 different types of common military armaments around the world posing threat to civilians on land."
|
61 |
+
)
|
62 |
iface.launch(inline=False) # share=True for colab
|