Spaces:
Running
Running
Fix bug and new samples
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def handle_classify(image=None):
|
|
30 |
|
31 |
|
32 |
inputs = [
|
33 |
-
gr.Image(
|
34 |
]
|
35 |
|
36 |
|
@@ -39,7 +39,7 @@ outputs = gr.Textbox()
|
|
39 |
title = "Racist model v2"
|
40 |
|
41 |
SAMPLE_DIR = 'samples'
|
42 |
-
examples = [
|
43 |
|
44 |
yolo_app = gr.Interface(
|
45 |
fn=handle_classify,
|
|
|
30 |
|
31 |
|
32 |
inputs = [
|
33 |
+
gr.Image(label="Input Image"),
|
34 |
]
|
35 |
|
36 |
|
|
|
39 |
title = "Racist model v2"
|
40 |
|
41 |
SAMPLE_DIR = 'samples'
|
42 |
+
examples = [path for path in os.listdir(SAMPLE_DIR)]
|
43 |
|
44 |
yolo_app = gr.Interface(
|
45 |
fn=handle_classify,
|