kk

app.py

@@ -1,56 +1,51 @@
-#Importing necessary libraries
 import gradio as gr
 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")  
+# Model functions
+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):
     input, target = input.cpu().numpy()[:,1], target.cpu().numpy()
     return roc_auc_score(target, input)
 
-#Load the model
-learn = load_learner("export.pkl")
+# Load model
+learn = load_learner("export.pkl")   
 
-#Identify labels from the dataloaders class  
+# Labels   
 labels = ["Negative", "Positive"]
 
-#Define function for making prediction
+# Prediction function
 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))}
 
-#Custom CSS    
+# Custom CSS
 custom_css = """
-body, button {
-  background-color: #00FBB9;  
-}
-
 button {
-  color: #333;  
-}
-
-p, span {
+  background-color: #00fbb9;
   color: #333;
+  border: none;
+  padding: 10px 20px;
+  border-radius: 5px;
 }
-"""
 
-#Customizing the gradio interface
-title = "Ethiopian based Tuberculosis(TB) model"
-description = """Model to detect TB from chest x-rays""" 
+button:hover {
+  background-color: #00d6a7;
+}  
+"""
 
+# Interface parameters
+title = "Ethiopia TB Detection"
+description = "Detect TB from chest x-rays"
 examples = ['patient1.png', 'patient2.png', 'patient3.png']
-enable_queue=True
 
-#Launching the gradio interface
-gr.Interface(fn=predict, 
-             inputs=gr.inputs.Image(shape=(512, 512)),
-             outputs=gr.outputs.Label(num_top_classes=1),  
+# Launch interface
+gr.Interface(fn=predict,
+             inputs=gr.inputs.Image(shape=(512, 512)), 
+             outputs=gr.outputs.Label(num_top_classes=1),
              title=title,
              description=description,
              examples=examples,
-             enable_queue=enable_queue,
              css=custom_css).launch(inline=False)