csuhan commited on
Commit
519f2af
1 Parent(s): 989f17e
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -40,7 +40,9 @@ def inference(input, model):
40
  output = visualized_output.get_image()[:, :, ::-1]
41
  return output
42
 
43
- examples = [[os.path.join("demo", x), "OpenDet-R50"] for x in os.listdir('demo') if x.endswith(".jpg")]
 
 
44
 
45
  iface = gr.Interface(
46
  inference,
 
40
  output = visualized_output.get_image()[:, :, ::-1]
41
  return output
42
 
43
+ examples = [
44
+ ["demo/000000002157.jpg", "OpenDet-R50"],
45
+ ["demo/000000020059.jpg", "OpenDet-R50"]]
46
 
47
  iface = gr.Interface(
48
  inference,