zihaoz96 commited on
Commit
b1cde34
1 Parent(s): 6d0987e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def predict_image(image, model_name):
58
  image = gr.inputs.Image(shape=(300, 300), label="Upload Your Image Here")
59
  label = gr.outputs.Label(num_top_classes=len(labels))
60
 
61
- samples = [["samples/" + p + "/jpg"] for p in labels]
62
 
63
  interface = gr.Interface(
64
  fn=predict_image,
 
58
  image = gr.inputs.Image(shape=(300, 300), label="Upload Your Image Here")
59
  label = gr.outputs.Label(num_top_classes=len(labels))
60
 
61
+ samples = [["samples/" + p + ".jpg"] for p in labels]
62
 
63
  interface = gr.Interface(
64
  fn=predict_image,