harishtaa commited on
Commit
4eb5719
·
1 Parent(s): 6d74711

Add application file

Browse files
Files changed (5) hide show
  1. App.py +22 -15
  2. import json.py +105 -0
  3. label.json +14 -0
  4. model.h5 +3 -0
  5. requirements.txt +1 -0
App.py CHANGED
@@ -1,22 +1,29 @@
 
1
  import gradio as gr
2
  import tensorflow as tf
3
  import numpy as np
4
- import requests
5
- from keras.applications.inception_v3 import preprocess_input
6
-
7
- inception_net = tf.keras.applications.InceptionV3()
8
 
9
- def classify_image(img):
10
- img_new = img.reshape((-1, 224, 224, 3))
11
- img = tf.keras.applications.inception_v3.preprocess_input(img_new)
12
- prediction = model.predict(img).flatten()
13
 
14
- return {class_names[i]: float(prediction[i]) for i in range(100)}
15
-
16
- image = gr.inputs.Image()
17
- gr.Image(source="webcam", streaming=True)
 
 
 
18
 
19
- label = gr.outputs.Label(num_top_classes= 5)
 
 
 
 
 
20
 
21
- my_project = gr.Interface(fn=classify_image, inputs=gr.Image(shape=(224, 224), source="upload", streaming=True), outputs=label)
22
- my_project.launch(debug=True, share=True)
 
1
+
2
  import gradio as gr
3
  import tensorflow as tf
4
  import numpy as np
5
+ import json
6
+ from os.path import dirname, realpath, join
 
 
7
 
8
+ # Load human-readable labels for ImageNet.
9
+ current_dir = dirname(realpath(__file__))
10
+ with open(join(current_dir, "label.json")) as labels_file:
11
+ labels = json.load(labels_file)
12
 
13
+ model = './model.h5'
14
+ inception_net = tf.keras.applications.InceptionV3()
15
+ def image_classifier(im):
16
+ arr = np.expand_dims(im, axis=0)
17
+ arr = tf.keras.applications.inception_v3.preprocess_input(arr)
18
+ prediction = model.predict(arr).flatten()
19
+ return {labels[i]: float(prediction[i]) for i in range(100)}
20
 
21
+ iface = gr.Interface(
22
+ image_classifier,
23
+ gr.inputs.Image(shape=(224, 224)),
24
+ gr.outputs.Label(num_top_classes=3),
25
+ capture_session=True,
26
+ interpretation="default")
27
 
28
+ if __name__ == "__main__":
29
+ iface.launch(share=True)
import json.py ADDED
@@ -0,0 +1,105 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import json
2
+
3
+
4
+ list = ['ADONIS',
5
+ 'AFRICAN GIANT SWALLOWTAIL',
6
+ 'AMERICAN SNOOT',
7
+ 'AN 88',
8
+ 'APPOLLO',
9
+ 'ARCIGERA FLOWER MOTH',
10
+ 'ATALA',
11
+ 'ATLAS MOTH',
12
+ 'BANDED ORANGE HELICONIAN',
13
+ 'BANDED PEACOCK',
14
+ 'BANDED TIGER MOTH',
15
+ 'BECKERS WHITE',
16
+ 'BIRD CHERRY ERMINE MOTH',
17
+ 'BLACK HAIRSTREAK',
18
+ 'BLUE MORPHO',
19
+ 'BLUE SPOTTED CROW',
20
+ 'BROOKES BIRDWING',
21
+ 'BROWN ARGUS',
22
+ 'BROWN SIPROETA',
23
+ 'CABBAGE WHITE',
24
+ 'CAIRNS BIRDWING',
25
+ 'CHALK HILL BLUE',
26
+ 'CHECQUERED SKIPPER',
27
+ 'CHESTNUT',
28
+ 'CINNABAR MOTH',
29
+ 'CLEARWING MOTH',
30
+ 'CLEOPATRA',
31
+ 'CLODIUS PARNASSIAN',
32
+ 'CLOUDED SULPHUR',
33
+ 'COMET MOTH',
34
+ 'COMMON BANDED AWL',
35
+ 'COMMON WOOD-NYMPH',
36
+ 'COPPER TAIL',
37
+ 'CRECENT',
38
+ 'CRIMSON PATCH',
39
+ 'DANAID EGGFLY',
40
+ 'EASTERN COMA',
41
+ 'EASTERN DAPPLE WHITE',
42
+ 'EASTERN PINE ELFIN',
43
+ 'ELBOWED PIERROT',
44
+ 'EMPEROR GUM MOTH',
45
+ 'GARDEN TIGER MOTH',
46
+ 'GIANT LEOPARD MOTH',
47
+ 'GLITTERING SAPPHIRE',
48
+ 'GOLD BANDED',
49
+ 'GREAT EGGFLY',
50
+ 'GREAT JAY',
51
+ 'GREEN CELLED CATTLEHEART',
52
+ 'GREEN HAIRSTREAK',
53
+ 'GREY HAIRSTREAK',
54
+ 'HERCULES MOTH',
55
+ 'HUMMING BIRD HAWK MOTH',
56
+ 'INDRA SWALLOW',
57
+ 'IO MOTH',
58
+ 'Iphiclus sister',
59
+ 'JULIA',
60
+ 'LARGE MARBLE',
61
+ 'LUNA MOTH',
62
+ 'MADAGASCAN SUNSET MOTH',
63
+ 'MALACHITE',
64
+ 'MANGROVE SKIPPER',
65
+ 'MESTRA',
66
+ 'METALMARK',
67
+ 'MILBERTS TORTOISESHELL',
68
+ 'MONARCH',
69
+ 'MOURNING CLOAK',
70
+ 'OLEANDER HAWK MOTH',
71
+ 'ORANGE OAKLEAF',
72
+ 'ORANGE TIP',
73
+ 'ORCHARD SWALLOW',
74
+ 'PAINTED LADY',
75
+ 'PAPER KITE',
76
+ 'PEACOCK',
77
+ 'PINE WHITE',
78
+ 'PIPEVINE SWALLOW',
79
+ 'POLYPHEMUS MOTH',
80
+ 'POPINJAY',
81
+ 'PURPLE HAIRSTREAK',
82
+ 'PURPLISH COPPER',
83
+ 'QUESTION MARK',
84
+ 'RED ADMIRAL',
85
+ 'RED CRACKER',
86
+ 'RED POSTMAN',
87
+ 'RED SPOTTED PURPLE',
88
+ 'ROSY MAPLE MOTH',
89
+ 'SCARCE SWALLOW',
90
+ 'SILVER SPOT SKIPPER',
91
+ 'SIXSPOT BURNET MOTH',
92
+ 'SLEEPY ORANGE',
93
+ 'SOOTYWING',
94
+ 'SOUTHERN DOGFACE',
95
+ 'STRAITED QUEEN',
96
+ 'TROPICAL LEAFWING',
97
+ 'TWO BARRED FLASHER',
98
+ 'ULYSES',
99
+ 'VICEROY',
100
+ 'WHITE LINED SPHINX MOTH',
101
+ 'WOOD SATYR',
102
+ 'YELLOW SWALLOW TAIL',
103
+ 'ZEBRA LONG WING']
104
+
105
+ jsonString = json.dumps(list)
label.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ["ADONIS",
2
+ "AFRICAN GIANT SWALLOWTAIL",
3
+ "AMERICAN SNOOT",
4
+ "AN 88",
5
+ "APPOLLO",
6
+ "ARCIGERA FLOWER MOTH",
7
+ "ATALA",
8
+ "ATLAS MOTH",
9
+ "BANDED ORANGE HELICONIAN",
10
+ "BANDED PEACOCK",
11
+ "BANDED TIGER MOTH",
12
+ "BECKERS WHITE",
13
+ "BIRD CHERRY ERMINE MOTH",
14
+ "BLACK HAIRSTREAK", "BLUE MORPHO", "BLUE SPOTTED CROW", "BROOKES BIRDWING", "BROWN ARGUS", "BROWN SIPROETA", "CABBAGE WHITE", "CAIRNS BIRDWING", "CHALK HILL BLUE", "CHECQUERED SKIPPER", "CHESTNUT", "CINNABAR MOTH", "CLEARWING MOTH", "CLEOPATRA", "CLODIUS PARNASSIAN", "CLOUDED SULPHUR", "COMET MOTH", "COMMON BANDED AWL", "COMMON WOOD-NYMPH", "COPPER TAIL", "CRECENT", "CRIMSON PATCH", "DANAID EGGFLY", "EASTERN COMA", "EASTERN DAPPLE WHITE", "EASTERN PINE ELFIN", "ELBOWED PIERROT", "EMPEROR GUM MOTH", "GARDEN TIGER MOTH", "GIANT LEOPARD MOTH", "GLITTERING SAPPHIRE", "GOLD BANDED", "GREAT EGGFLY", "GREAT JAY", "GREEN CELLED CATTLEHEART", "GREEN HAIRSTREAK", "GREY HAIRSTREAK", "HERCULES MOTH", "HUMMING BIRD HAWK MOTH", "INDRA SWALLOW", "IO MOTH", "Iphiclus sister", "JULIA", "LARGE MARBLE", "LUNA MOTH", "MADAGASCAN SUNSET MOTH", "MALACHITE", "MANGROVE SKIPPER", "MESTRA", "METALMARK", "MILBERTS TORTOISESHELL", "MONARCH", "MOURNING CLOAK", "OLEANDER HAWK MOTH", "ORANGE OAKLEAF", "ORANGE TIP", "ORCHARD SWALLOW", "PAINTED LADY", "PAPER KITE", "PEACOCK", "PINE WHITE", "PIPEVINE SWALLOW", "POLYPHEMUS MOTH", "POPINJAY", "PURPLE HAIRSTREAK", "PURPLISH COPPER", "QUESTION MARK", "RED ADMIRAL", "RED CRACKER", "RED POSTMAN", "RED SPOTTED PURPLE", "ROSY MAPLE MOTH", "SCARCE SWALLOW", "SILVER SPOT SKIPPER", "SIXSPOT BURNET MOTH", "SLEEPY ORANGE", "SOOTYWING", "SOUTHERN DOGFACE", "STRAITED QUEEN", "TROPICAL LEAFWING", "TWO BARRED FLASHER", "ULYSES", "VICEROY", "WHITE LINED SPHINX MOTH", "WOOD SATYR", "YELLOW SWALLOW TAIL", "ZEBRA LONG WING"]["ADONIS", "AFRICAN GIANT SWALLOWTAIL", "AMERICAN SNOOT", "AN 88", "APPOLLO", "ARCIGERA FLOWER MOTH", "ATALA", "ATLAS MOTH", "BANDED ORANGE HELICONIAN", "BANDED PEACOCK", "BANDED TIGER MOTH", "BECKERS WHITE", "BIRD CHERRY ERMINE MOTH", "BLACK HAIRSTREAK", "BLUE MORPHO", "BLUE SPOTTED CROW", "BROOKES BIRDWING", "BROWN ARGUS", "BROWN SIPROETA", "CABBAGE WHITE", "CAIRNS BIRDWING", "CHALK HILL BLUE", "CHECQUERED SKIPPER", "CHESTNUT", "CINNABAR MOTH", "CLEARWING MOTH", "CLEOPATRA", "CLODIUS PARNASSIAN", "CLOUDED SULPHUR", "COMET MOTH", "COMMON BANDED AWL", "COMMON WOOD-NYMPH", "COPPER TAIL", "CRECENT", "CRIMSON PATCH", "DANAID EGGFLY", "EASTERN COMA", "EASTERN DAPPLE WHITE", "EASTERN PINE ELFIN", "ELBOWED PIERROT", "EMPEROR GUM MOTH", "GARDEN TIGER MOTH", "GIANT LEOPARD MOTH", "GLITTERING SAPPHIRE", "GOLD BANDED", "GREAT EGGFLY", "GREAT JAY", "GREEN CELLED CATTLEHEART", "GREEN HAIRSTREAK", "GREY HAIRSTREAK", "HERCULES MOTH", "HUMMING BIRD HAWK MOTH", "INDRA SWALLOW", "IO MOTH", "Iphiclus sister", "JULIA", "LARGE MARBLE", "LUNA MOTH", "MADAGASCAN SUNSET MOTH", "MALACHITE", "MANGROVE SKIPPER", "MESTRA", "METALMARK", "MILBERTS TORTOISESHELL", "MONARCH", "MOURNING CLOAK", "OLEANDER HAWK MOTH", "ORANGE OAKLEAF", "ORANGE TIP", "ORCHARD SWALLOW", "PAINTED LADY", "PAPER KITE", "PEACOCK", "PINE WHITE", "PIPEVINE SWALLOW", "POLYPHEMUS MOTH", "POPINJAY", "PURPLE HAIRSTREAK", "PURPLISH COPPER", "QUESTION MARK", "RED ADMIRAL", "RED CRACKER", "RED POSTMAN", "RED SPOTTED PURPLE", "ROSY MAPLE MOTH", "SCARCE SWALLOW", "SILVER SPOT SKIPPER", "SIXSPOT BURNET MOTH", "SLEEPY ORANGE", "SOOTYWING", "SOUTHERN DOGFACE", "STRAITED QUEEN", "TROPICAL LEAFWING", "TWO BARRED FLASHER", "ULYSES", "VICEROY", "WHITE LINED SPHINX MOTH", "WOOD SATYR", "YELLOW SWALLOW TAIL", "ZEBRA LONG WING"]["ADONIS", "AFRICAN GIANT SWALLOWTAIL", "AMERICAN SNOOT", "AN 88", "APPOLLO", "ARCIGERA FLOWER MOTH", "ATALA", "ATLAS MOTH", "BANDED ORANGE HELICONIAN", "BANDED PEACOCK", "BANDED TIGER MOTH", "BECKERS WHITE", "BIRD CHERRY ERMINE MOTH", "BLACK HAIRSTREAK", "BLUE MORPHO", "BLUE SPOTTED CROW", "BROOKES BIRDWING", "BROWN ARGUS", "BROWN SIPROETA", "CABBAGE WHITE", "CAIRNS BIRDWING", "CHALK HILL BLUE", "CHECQUERED SKIPPER", "CHESTNUT", "CINNABAR MOTH", "CLEARWING MOTH", "CLEOPATRA", "CLODIUS PARNASSIAN", "CLOUDED SULPHUR", "COMET MOTH", "COMMON BANDED AWL", "COMMON WOOD-NYMPH", "COPPER TAIL", "CRECENT", "CRIMSON PATCH", "DANAID EGGFLY", "EASTERN COMA", "EASTERN DAPPLE WHITE", "EASTERN PINE ELFIN", "ELBOWED PIERROT", "EMPEROR GUM MOTH", "GARDEN TIGER MOTH", "GIANT LEOPARD MOTH", "GLITTERING SAPPHIRE", "GOLD BANDED", "GREAT EGGFLY", "GREAT JAY", "GREEN CELLED CATTLEHEART", "GREEN HAIRSTREAK", "GREY HAIRSTREAK", "HERCULES MOTH", "HUMMING BIRD HAWK MOTH", "INDRA SWALLOW", "IO MOTH", "Iphiclus sister", "JULIA", "LARGE MARBLE", "LUNA MOTH", "MADAGASCAN SUNSET MOTH", "MALACHITE", "MANGROVE SKIPPER", "MESTRA", "METALMARK", "MILBERTS TORTOISESHELL", "MONARCH", "MOURNING CLOAK", "OLEANDER HAWK MOTH", "ORANGE OAKLEAF", "ORANGE TIP", "ORCHARD SWALLOW", "PAINTED LADY", "PAPER KITE", "PEACOCK", "PINE WHITE", "PIPEVINE SWALLOW", "POLYPHEMUS MOTH", "POPINJAY", "PURPLE HAIRSTREAK", "PURPLISH COPPER", "QUESTION MARK", "RED ADMIRAL", "RED CRACKER", "RED POSTMAN", "RED SPOTTED PURPLE", "ROSY MAPLE MOTH", "SCARCE SWALLOW", "SILVER SPOT SKIPPER", "SIXSPOT BURNET MOTH", "SLEEPY ORANGE", "SOOTYWING", "SOUTHERN DOGFACE", "STRAITED QUEEN", "TROPICAL LEAFWING", "TWO BARRED FLASHER", "ULYSES", "VICEROY", "WHITE LINED SPHINX MOTH", "WOOD SATYR", "YELLOW SWALLOW TAIL", "ZEBRA LONG WING"]["ADONIS", "AFRICAN GIANT SWALLOWTAIL", "AMERICAN SNOOT", "AN 88", "APPOLLO", "ARCIGERA FLOWER MOTH", "ATALA", "ATLAS MOTH", "BANDED ORANGE HELICONIAN", "BANDED PEACOCK", "BANDED TIGER MOTH", "BECKERS WHITE", "BIRD CHERRY ERMINE MOTH", "BLACK HAIRSTREAK", "BLUE MORPHO", "BLUE SPOTTED CROW", "BROOKES BIRDWING", "BROWN ARGUS", "BROWN SIPROETA", "CABBAGE WHITE", "CAIRNS BIRDWING", "CHALK HILL BLUE", "CHECQUERED SKIPPER", "CHESTNUT", "CINNABAR MOTH", "CLEARWING MOTH", "CLEOPATRA", "CLODIUS PARNASSIAN", "CLOUDED SULPHUR", "COMET MOTH", "COMMON BANDED AWL", "COMMON WOOD-NYMPH", "COPPER TAIL", "CRECENT", "CRIMSON PATCH", "DANAID EGGFLY", "EASTERN COMA", "EASTERN DAPPLE WHITE", "EASTERN PINE ELFIN", "ELBOWED PIERROT", "EMPEROR GUM MOTH", "GARDEN TIGER MOTH", "GIANT LEOPARD MOTH", "GLITTERING SAPPHIRE", "GOLD BANDED", "GREAT EGGFLY", "GREAT JAY", "GREEN CELLED CATTLEHEART", "GREEN HAIRSTREAK", "GREY HAIRSTREAK", "HERCULES MOTH", "HUMMING BIRD HAWK MOTH", "INDRA SWALLOW", "IO MOTH", "Iphiclus sister", "JULIA", "LARGE MARBLE", "LUNA MOTH", "MADAGASCAN SUNSET MOTH", "MALACHITE", "MANGROVE SKIPPER", "MESTRA", "METALMARK", "MILBERTS TORTOISESHELL", "MONARCH", "MOURNING CLOAK", "OLEANDER HAWK MOTH", "ORANGE OAKLEAF", "ORANGE TIP", "ORCHARD SWALLOW", "PAINTED LADY", "PAPER KITE", "PEACOCK", "PINE WHITE", "PIPEVINE SWALLOW", "POLYPHEMUS MOTH", "POPINJAY", "PURPLE HAIRSTREAK", "PURPLISH COPPER", "QUESTION MARK", "RED ADMIRAL", "RED CRACKER", "RED POSTMAN", "RED SPOTTED PURPLE", "ROSY MAPLE MOTH", "SCARCE SWALLOW", "SILVER SPOT SKIPPER", "SIXSPOT BURNET MOTH", "SLEEPY ORANGE", "SOOTYWING", "SOUTHERN DOGFACE", "STRAITED QUEEN", "TROPICAL LEAFWING", "TWO BARRED FLASHER", "ULYSES", "VICEROY", "WHITE LINED SPHINX MOTH", "WOOD SATYR", "YELLOW SWALLOW TAIL", "ZEBRA LONG WING"]
model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:acaacb559e18aa9e9a1300dc3807839e9d97814b9d5856df84919f375f6baab2
3
+ size 167545288
requirements.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ tensorflow