ducngg commited on
Commit
27cf630
1 Parent(s): fca9a42

Fix bug and new samples

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ def handle_classify(image=None):
30
 
31
 
32
  inputs = [
33
- gr.Image(type='numpy', label="Input Image"),
34
  ]
35
 
36
 
@@ -39,7 +39,7 @@ outputs = gr.Textbox()
39
  title = "Racist model v2"
40
 
41
  SAMPLE_DIR = 'samples'
42
- examples = [np.array(Image.open(os.path.join(SAMPLE_DIR, path))) for path in os.listdir(SAMPLE_DIR)]
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,