tanvir-ishraq commited on
Commit
b4a7d51
1 Parent(s): c860493

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -4
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', 'Heron TP drone',
 
18
  'Humvee vehicle',
19
  'LAV-25 vehicle',
20
  'Leopard 2 tank',
21
  'M1 Abrams tank',
22
  'M109 artillery',
23
- 'M113 vehicle', 'M270 MLRS artillery',
 
24
  'MQ-9 Reaper UVAC drone',
25
  'MRAP vehicle',
26
  'RQ-4 Global Hawk UVAC drone',
27
  'T-72 tank',
28
- 'Type 99 tank', 'smerch artillery'
 
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(fn=recognize_image, inputs=image, outputs=label, examples=examples)
 
 
 
 
 
 
 
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