ttheland commited on
Commit
30dbf3f
1 Parent(s): ca25ee1

theming and outputs change

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -83,6 +83,7 @@ def do_inference(img):
83
  im = gr.inputs.Image(shape=(512, 512), image_mode='RGB',
84
  invert_colors=False, source="upload",
85
  type="pil")
 
86
  title = "Butterfly Classification Demo"
87
  description = "A pretrained ResNet18 CNN trained on the Leeds Butterfly Dataset. Libraries: PyTorch, Gradio."
88
  examples = [['./h_erato.jpg'],['d_plexippus.jpg']]
@@ -113,14 +114,16 @@ enable_queue = True
113
  iface = gr.Interface(
114
  do_inference,
115
  im,
116
- gr.outputs.Label(num_top_classes=2),
117
  live=False,
118
  interpretation=None,
119
  title=title,
120
  description=description,
121
  article= article_text,
122
  examples=examples,
123
- enable_queue=enable_queue
 
 
124
  )
125
 
126
  iface.test_launch()
 
83
  im = gr.inputs.Image(shape=(512, 512), image_mode='RGB',
84
  invert_colors=False, source="upload",
85
  type="pil")
86
+
87
  title = "Butterfly Classification Demo"
88
  description = "A pretrained ResNet18 CNN trained on the Leeds Butterfly Dataset. Libraries: PyTorch, Gradio."
89
  examples = [['./h_erato.jpg'],['d_plexippus.jpg']]
 
114
  iface = gr.Interface(
115
  do_inference,
116
  im,
117
+ gr.outputs.Label(num_top_classes=3),
118
  live=False,
119
  interpretation=None,
120
  title=title,
121
  description=description,
122
  article= article_text,
123
  examples=examples,
124
+ enable_queue=enable_queue,
125
+ theme="dark-peach"
126
+
127
  )
128
 
129
  iface.test_launch()