jkjk

app.py

@@ -1,11 +1,10 @@
 #Importing necessary libraries
 import gradio as gr
-#import scikit-learn as sklearn
 from fastai.vision.all import *
 from sklearn.metrics import roc_auc_score
 
 #Define dependent functions
-def get_x(row): return Path(str(path/f"{row['rootname']}_small"/f"{row['ID']}") + ".png")
+def get_x(row): return Path(str(path/f"{row['rootname']}_small"/f"{row['ID']}") + ".png")  
 def get_y(row): return row["LABEL"]
 
 def auroc_score(input, target):
@@ -15,31 +14,43 @@ def auroc_score(input, target):
 #Load the model
 learn = load_learner("export.pkl")
 
-#Identify labels from the dataloaders class
+#Identify labels from the dataloaders class  
 labels = ["Negative", "Positive"]
 
 #Define function for making prediction
 def predict(img):
-    img = PILImage.create(img)
+    img = PILImage.create(img) 
     pred, idx, probs = learn.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
-#Customizing the gradio interface
-title = "Ethiopian based Tuberculosis(TB) model"
+#Custom CSS    
+custom_css = """
+body, button {
+  background-color: #00FBB9;  
+}
 
-description = """e"""
+button {
+  color: #333;  
+}
 
-article="<p style='text-align: center'><a href='https' target='_blank'>n</a></p>"
+p, span {
+  color: #333;
+}
+"""
 
-examples = ['patient1.png', 'patient2.png', 'patient3.png']
+#Customizing the gradio interface
+title = "Ethiopian based Tuberculosis(TB) model"
+description = """Model to detect TB from chest x-rays""" 
 
+examples = ['patient1.png', 'patient2.png', 'patient3.png']
 enable_queue=True
 
-
-#Launching the gradio application
-gr.Interface(fn=predict,inputs=gr.inputs.Image(shape=(512, 512)),
-             outputs=gr.outputs.Label(num_top_classes=1),
+#Launching the gradio interface
+gr.Interface(fn=predict, 
+             inputs=gr.inputs.Image(shape=(512, 512)),
+             outputs=gr.outputs.Label(num_top_classes=1),  
              title=title,
-             description=description,article=article,
+             description=description,
              examples=examples,
-             enable_queue=enable_queue).launch(inline=False)
+             enable_queue=enable_queue,
+             css=custom_css).launch(inline=False)