zihaoz96 commited on
Commit
f447626
1 Parent(s): 57c2b2d
app.py CHANGED
@@ -57,6 +57,8 @@ def predict_image(image, model_name):
57
 
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
  interface = gr.Interface(
62
  fn=predict_image,
@@ -65,6 +67,7 @@ interface = gr.Interface(
65
  capture_session=True,
66
  allow_flagging=False,
67
  title=“🦈 Shark image classifier”,
68
- description="Made with HugsVision & ❤️"
 
69
  )
70
  interface.launch()
 
57
 
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,
 
67
  capture_session=True,
68
  allow_flagging=False,
69
  title=“🦈 Shark image classifier”,
70
+ description="Made with HugsVision & ❤️",
71
+ examples=samples
72
  )
73
  interface.launch()
samples/basking.jpg ADDED
samples/black tip.jpg ADDED
samples/blue.jpg ADDED
samples/bull.jpg ADDED
samples/hammer head.jpg ADDED
samples/lemon.jpg ADDED
samples/mako.jpg ADDED
samples/nurse.jpg ADDED
samples/sand tiger.jpg ADDED
samples/thresher.jpg ADDED
samples/tigre.jpg ADDED
samples/whale.jpg ADDED
samples/white tip.jpg ADDED
samples/white.jpg ADDED