fohy24 commited on
Commit
f7724ab
·
1 Parent(s): e25cd6e

update layout from gr.interface to gr.blocks; add accordion to show prediction labels

Browse files
Files changed (1) hide show
  1. app.py +30 -8
app.py CHANGED
@@ -82,13 +82,35 @@ def predict(img, confidence):
82
  return prediction_dict
83
 
84
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
85
 
86
- gr.Interface(fn=predict,
87
- inputs=[gr.Image(type="pil"),
88
- gr.Slider(0, 1, value=0.5, label="Confidence", info="Show predictions that are above this confidence level")],
89
- outputs=gr.Label(),
90
- examples=[["enchi_albino_clown.png", 0.5], ["mojave_ghi.png", 0.5], ["hypo_banana_pastel_enchi.png", 0.5]],
91
- title='Ball Python Morph Identifier',
92
- description="Upload or paste an image of your ball python to identify its morphs!"
93
- ).launch()
94
 
 
82
  return prediction_dict
83
 
84
 
85
+ with gr.Blocks(title='Ball Python Morph Identifier') as demo:
86
+ gr.Markdown("# Ball Python Morph Identifier")
87
+ gr.Markdown("Upload or paste an image of your ball python to identify its morphs!")
88
+
89
+ with gr.Accordion("Click here to show all the morphs that can be predicted", open=False):
90
+ gr.Markdown("""
91
+ Albino, Asphalt, Banana, Black Head, Black Pastel, Butter, Calico, Chocolate, Cinnamon, Clown, Desert Ghost, Enchi, Fire,
92
+ GHI, Gravel, Hypo, Leopard, Lesser, Mojave, Normal, Orange Dream, Pastel, Piebald, Pinstripe, Red Stripe,
93
+ Spider, Spotnose, Super Pastel, Vanilla, Yellow Belly
94
+ """)
95
+
96
+ with gr.Row():
97
+ with gr.Column(scale=1):
98
+ img_input = gr.Image(type="pil", label="Upload/Paste Image")
99
+ gr.Examples(
100
+ examples=[
101
+ ["enchi_albino_clown.png", "Enchi, Albino, Clown"],
102
+ ["mojave_ghi.png", "Mojave, GHI"],
103
+ ["hypo_banana_pastel_enchi.png", "Hypo, Banana, Pastel, Enchi"]],
104
+ inputs=[img_input]
105
+ )
106
+ confidence = gr.Slider(0, 1, value=0.5, label="Confidence",
107
+ info="Show predictions that are above this confidence level")
108
+ predict_btn = gr.Button("Identify Morphs", variant="primary")
109
+
110
+ with gr.Column(scale=1):
111
+ label_output = gr.Label(label="Predicted Morphs")
112
+
113
+ predict_btn.click(fn=predict, inputs=[img_input, confidence], outputs=label_output)
114
 
115
+ demo.launch()
 
 
 
 
 
 
 
116