abrar-adnan commited on
Commit
03af31e
·
1 Parent(s): 74f16fa

added files

Browse files
Files changed (1) hide show
  1. app.py +60 -0
app.py ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastai.vision.all import *
2
+ import gradio as gr
3
+
4
+ # import pathlib
5
+ # temp = pathlib.PosixPath
6
+ # pathlib.PosixPath = pathlib.WindowsPath
7
+
8
+ vehicle_labels = (
9
+ "Airplane",
10
+ "Ambulance",
11
+ "Armored Tank",
12
+ "ATV",
13
+ "Autorickshaw",
14
+ "Bicycle",
15
+ "Boat",
16
+ "Buggy",
17
+ "Bulldozer",
18
+ "Cargo Ship",
19
+ "Cargo Truck",
20
+ "Crane",
21
+ "Excavator",
22
+ "Ferry",
23
+ "Helicopter",
24
+ "Hot Air Baloon",
25
+ "Microbus",
26
+ "Monster Truck",
27
+ "Motorcycle",
28
+ "Multi Purpose Vehicle",
29
+ "Ocean Liner",
30
+ "Police Car",
31
+ "Private Car",
32
+ "Rickshaw",
33
+ "Sail Boat",
34
+ "Semi Truck",
35
+ "Sports Car",
36
+ "Steam Roller",
37
+ "SUV",
38
+ "Transport Bus",
39
+ "Train",
40
+ "Truck",
41
+ "Yacht"
42
+ )
43
+
44
+ model = load_learner('vehicle-recognizer-v1.pkl')
45
+
46
+ def recognize_image(image):
47
+ pred, idx, probs = model.predict(image)
48
+ return dict(zip(vehicle_labels, map(float, probs)))
49
+
50
+ image = gr.inputs.Image(shape=(192,192))
51
+ label = gr.outputs.Label(num_top_classes=5)
52
+ examples = [
53
+ 'unknown00.png',
54
+ 'unknown01.png',
55
+ 'unknown02.png',
56
+ 'unknown03.png'
57
+ ]
58
+
59
+ iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
60
+ iface.launch(inline=False)