tianyilexi commited on
Commit
91992e9
·
1 Parent(s): 78cff7e
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,10 +19,10 @@ def predict(img):
19
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
20
 
21
  title = "Northern EU Mushroom Classifier"
22
- description = "<p style='text-align: center, font-size:14px'>A North EU mushroom image classifier trained on a kaggle dataset with fastai. The dataset consist of 9 different folders that contains from 300 to 1500 selected images of mushrooms genuses.</p>"
23
- article="<p style='text-align: center, font-size:14px'><a href='https://www.kaggle.com/datasets/maysee/mushrooms-classification-common-genuss-images' target='_blank'>Data Source</a></h4>"
24
  interpretation='default'
25
  enable_queue=True
26
  inputs = gr.Image(shape=(224, 224))
27
 
28
- gr.Interface(fn=predict, inputs=inputs, outputs=gr.Label(num_top_classes=3), title=title, description=description, article=article, interpretation=interpretation, examples=examples).launch(inline=False, enable_queue=enable_queue)
 
19
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
20
 
21
  title = "Northern EU Mushroom Classifier"
22
+ description = "<p style='text-align: center; font-size:14px'>A North EU mushroom image classifier trained on a kaggle dataset with fastai. The dataset consist of 9 different folders that contains from 300 to 1500 selected images of mushrooms genuses.</p>"
23
+ article="<p style='text-align: center; font-size:14px'><a href='https://www.kaggle.com/datasets/maysee/mushrooms-classification-common-genuss-images' target='_blank'>Data Source</a></h4>"
24
  interpretation='default'
25
  enable_queue=True
26
  inputs = gr.Image(shape=(224, 224))
27
 
28
+ gr.Interface(fn=predict, inputs=inputs, outputs=gr.Label(num_top_classes=3), title=title, description=description, article=article, interpretation=interpretation, examples=examples).launch(inline=False, share=True, enable_queue=enable_queue)