Spaces:
Runtime error
Runtime error
Jason Adrian
commited on
Commit
·
6c52afe
1
Parent(s):
ef3c636
multiple images
Browse files
app.py
CHANGED
@@ -5,7 +5,8 @@ class_names = ['cat', 'dog']
|
|
5 |
|
6 |
def update_dropdown(className):
|
7 |
class_names.append(className)
|
8 |
-
|
|
|
9 |
|
10 |
def show_picked_class(className):
|
11 |
return className
|
@@ -30,6 +31,8 @@ def image_classifier(inp):
|
|
30 |
demo = gr.Blocks()
|
31 |
|
32 |
with demo as app:
|
|
|
|
|
33 |
with gr.Row():
|
34 |
with gr.Column():
|
35 |
inp_img = gr.Image()
|
@@ -57,5 +60,39 @@ with demo as app:
|
|
57 |
inputs=None,
|
58 |
outputs=[inp_img, out_txt])
|
59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
|
61 |
demo.launch(debug=True)
|
|
|
5 |
|
6 |
def update_dropdown(className):
|
7 |
class_names.append(className)
|
8 |
+
updated_choices = gr.Dropdown(choices=class_names)
|
9 |
+
return updated_choices
|
10 |
|
11 |
def show_picked_class(className):
|
12 |
return className
|
|
|
31 |
demo = gr.Blocks()
|
32 |
|
33 |
with demo as app:
|
34 |
+
gr.Markdown("# Single Image")
|
35 |
+
|
36 |
with gr.Row():
|
37 |
with gr.Column():
|
38 |
inp_img = gr.Image()
|
|
|
60 |
inputs=None,
|
61 |
outputs=[inp_img, out_txt])
|
62 |
|
63 |
+
gr.Markdown("# Multiple Images")
|
64 |
+
|
65 |
+
def show_to_gallery(images):
|
66 |
+
file_paths = [[file.name, class_names[0]] for file in images]
|
67 |
+
# print(file_paths)
|
68 |
+
return file_paths, file_paths
|
69 |
+
|
70 |
+
def get_select_index(evt: gr.SelectData):
|
71 |
+
# print("data",evt._data)
|
72 |
+
# print("value",evt.value)
|
73 |
+
return evt.index
|
74 |
+
|
75 |
+
with gr.Column():
|
76 |
+
imgs = gr.State()
|
77 |
+
|
78 |
+
multiple_inputs = gr.UploadButton(label="Upload multiple images file here.", file_count="multiple", file_types=["image"])
|
79 |
+
gallery = gr.Gallery()
|
80 |
+
selected = gr.Textbox(label="Image Gallery Index")
|
81 |
+
|
82 |
+
images_label = gr.Dropdown(class_names, label="Class Label", multiselect=False)
|
83 |
+
b3 = gr.Button("Save and change the label using dropdown")
|
84 |
+
|
85 |
+
multiple_inputs.upload(show_to_gallery, inputs=multiple_inputs, outputs=[gallery, imgs])
|
86 |
+
|
87 |
+
gallery.select(get_select_index, None, selected)
|
88 |
+
|
89 |
+
def change_labels(imgs, index, images_label):
|
90 |
+
index = int(index)
|
91 |
+
label_idx = class_names.index(images_label)
|
92 |
+
imgs[index][1] = class_names[label_idx]
|
93 |
+
return imgs, imgs
|
94 |
+
|
95 |
+
b3.click(change_labels, [imgs, selected, images_label], [imgs, gallery])
|
96 |
+
|
97 |
|
98 |
demo.launch(debug=True)
|