Update app.py

app.py

@@ -68,6 +68,14 @@ def predict_disease(image):
     }
     return result
 
+# References to display
+references = """
+1. Huang, G., et al. (2017). Densely Connected Convolutional Networks. Proceedings of the IEEE conference on computer vision and pattern recognition.
+2. Wang, X., et al. (2017). ChestX-ray8: Hospital-scale chest X-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases. IEEE CVPR.
+3. Rajpurkar, P., et al. (2017). CheXNet: Radiologist-level pneumonia detection on chest X-rays with deep learning. arXiv preprint arXiv:1711.05225.
+4. Abid, A., et al. (2019). Gradio: Hassle-Free Sharing and Testing of Machine Learning Models. arXiv preprint arXiv:1906.02569.
+"""
+
 # Gradio Interface
 interface = gr.Interface(
     fn=predict_disease,
@@ -75,6 +83,7 @@ interface = gr.Interface(
     outputs="label",  # Output is a dictionary of labels with probabilities
     title="CheXNet Pneumonia Detection",
     description="Upload a chest X-ray to detect the probability of 14 different diseases.",
+    additional_outputs=[gr.components.Textbox(label="References", value=references, lines=10)],  # Display references
 )
 
 # Launch the Gradio app