Update app.py

app.py

@@ -31,16 +31,27 @@ def classify_image(image):
 
     return label
 
-# Create title, description and article strings
-title = "Classification Demo"
-description = "XRay classification"
-
 # Create Gradio interface
-iface = gr.Interface(fn=classify_image, 
-                     inputs=gr.Image(),  # Accepts image of any size
-                     outputs=gr.Label(),
-                     title=title,
-                     description=description)
+def make_block(dem):
+    with dem:
+        gr.Markdown("Medical - Lungs Disease Prediction")
+        with gr.Tabs():
+            with gr.TabItem("Pneumonia Detection"):
+                with gr.Row():
+                    in_prompt_1 = gr.Image()
+                    out_response_1 = gr.Label()
+                b1 = gr.Button("Enter")
+
+            with gr.TabItem("Tuberculosis Detection"):
+                with gr.Row():
+                    in_prompt_2 = gr.Image()
+                    out_response_2 = gr.Label()
+                b2 = gr.Button("Enter")
+            b1.click(predict_sentiment, inputs=in_prompt_1, outputs=out_response_1)
+            b2.click(predict, inputs=in_prompt_2, outputs=out_response_2)
+           
+if __name__ == '__main__':
 
-# Launch the app
-iface.launch()
+    demo = gr.Blocks()
+    make_block(demo)
+    demo.launch()