YonaniCodes commited on
Commit
b3e1e17
·
1 Parent(s): a91136c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,13 +70,13 @@ def predict_breed(image):
70
  # Draw the actual text
71
  draw.text(text_position, text, font=font, fill=text_color)
72
 
73
- return img_with_text, f"{predicted_class} ({confidence:.2f}%)"
74
 
75
  # Create Gradio interface
76
  interface = gr.Interface(
77
  fn=predict_breed,
78
  inputs=gr.Image(type="pil"),
79
- outputs=[gr.Image(type="pil"), gr.Text()],
80
  title="Dog Breed Identifier 🐶",
81
  description="Upload a dog image and the model will predict the breed along with confidence!"
82
  )
 
70
  # Draw the actual text
71
  draw.text(text_position, text, font=font, fill=text_color)
72
 
73
+ return f"{predicted_class} ({confidence:.2f}%)"
74
 
75
  # Create Gradio interface
76
  interface = gr.Interface(
77
  fn=predict_breed,
78
  inputs=gr.Image(type="pil"),
79
+ outputs=gr.Text(),
80
  title="Dog Breed Identifier 🐶",
81
  description="Upload a dog image and the model will predict the breed along with confidence!"
82
  )