Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import ViTForImageClassification, ViTFeatureExtractor
 from PIL import Image
 import torch
 import numpy as np
+from gradio.themes import Theme
 
 # Load the pre-trained model and preprocessor (feature extractor)
 model_name_pneumonia = "runaksh/chest_xray_pneumonia_detection"
@@ -36,8 +37,10 @@ def classify_image(image):
 
     return label
 
+custom_theme = Theme(text_size="1.8em")
+
 # Create title, description and article strings
-title = "Classification Demo"
+title = "Automated Classification of Pneumonia and Tuberculosis using Machine Learning"
 description = "XRay classification"
 
 css_code = """
@@ -52,8 +55,10 @@ iface = gr.Interface(fn=classify_image,
                      outputs=gr.Label(),
                      title=title,
                      description=description,
-                     css=css_code
+                     css=css_code,
+                     theme=custom_theme
                     )
 
 # Launch the app 
 iface.launch()
+