Mohammed Innat commited on
Commit
42f5de6
1 Parent(s): 474f780

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -83,7 +83,7 @@ iface = gr.Interface(
83
  gr.inputs.Image(label="Transformer GradCAM"),
84
  ],
85
  title="Hybrid EfficientNet Swin Transformer Demo",
86
- description="The model is trained on tf_flowers dataset.",
87
  examples=[
88
  ["examples/dandelion.jpg"],
89
  ["examples/sunflower.jpg"],
@@ -93,3 +93,5 @@ iface = gr.Interface(
93
  ],
94
  )
95
  iface.launch()
 
 
83
  gr.inputs.Image(label="Transformer GradCAM"),
84
  ],
85
  title="Hybrid EfficientNet Swin Transformer Demo",
86
+ description="The model is trained on tf_flowers dataset <a href='https://www.kaggle.com/datasets/alxmamaev/flowers-recognition'>Flowers Recognition Dataset</a>. It provides 5 categories, namely: `daisy`, `rose`, `sunflower`, `tulip`, `dandelion`. One example from each class is provided in the Example section.",
87
  examples=[
88
  ["examples/dandelion.jpg"],
89
  ["examples/sunflower.jpg"],
93
  ],
94
  )
95
  iface.launch()
96
+
97
+