SagarPuniyani commited on
Commit
2ec730e
·
verified ·
1 Parent(s): 207bbf3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -46,11 +46,15 @@ def classify_image(image):
46
  interface = gr.Interface(
47
  fn=classify_image,
48
  inputs=gr.Image(type="pil"),
49
- outputs= f'{gr.Label(num_top_classes=3)} cancer stage',
 
 
 
50
  title="EfficientNet B3 Image Classifier",
51
  description="Upload an image to classify using the trained EfficientNet B3 model."
52
  )
53
 
 
54
  # Launch the app
55
  if __name__ == "__main__":
56
  interface.launch(share=True)
 
46
  interface = gr.Interface(
47
  fn=classify_image,
48
  inputs=gr.Image(type="pil"),
49
+ outputs=[
50
+ gr.Label(num_top_classes=3),
51
+ gr.Textbox(value="The prediction results above represent the top 3 cancer stages."),
52
+ ],
53
  title="EfficientNet B3 Image Classifier",
54
  description="Upload an image to classify using the trained EfficientNet B3 model."
55
  )
56
 
57
+
58
  # Launch the app
59
  if __name__ == "__main__":
60
  interface.launch(share=True)