enesmanan commited on
Commit
1082a29
·
verified ·
1 Parent(s): d13bccc

add class description

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -218,6 +218,7 @@ class AnimalClassifierApp:
218
  return [fig, text_results]
219
 
220
  def create_interface(self):
 
221
  return gr.Interface(
222
  fn=self.predict,
223
  inputs=gr.Image(type="pil"),
@@ -226,7 +227,11 @@ class AnimalClassifierApp:
226
  gr.Textbox(label="Detailed Results", lines=10)
227
  ],
228
  title="Animal Classifier - Model Comparison",
229
- description="Upload an image of an animal to see predictions from both EfficientNet and CNN models."
 
 
 
 
230
  )
231
 
232
 
 
218
  return [fig, text_results]
219
 
220
  def create_interface(self):
221
+ """Create Gradio interface."""
222
  return gr.Interface(
223
  fn=self.predict,
224
  inputs=gr.Image(type="pil"),
 
227
  gr.Textbox(label="Detailed Results", lines=10)
228
  ],
229
  title="Animal Classifier - Model Comparison",
230
+ description=(
231
+ "Upload an image of one of these animals: Bird, Cat, Dog, or Horse.\n"
232
+ "The app will compare predictions from both EfficientNet and CNN models.\n\n"
233
+ "Note: For best results, ensure the animal is clearly visible in the image."
234
+ )
235
  )
236
 
237