initial commit

app.py

@@ -0,0 +1,88 @@
+import gradio as gr
+import torch
+from torchvision.transforms import transforms
+import numpy as np
+from typing import Optional
+import torch.nn as nn
+import os
+from utils import page_utils
+
+from ultralytics import YOLO
+
+# Load a model
+model = YOLO('model_- 14 december 2023 12_01.pt')  # pretrained YOLOv8n model
+
+class_names = ['abdominal', 'adult', 'others', 'pediatric', 'spine']
+class_names.sort()
+
+examples_dir = "samples"
+
+def image_classifier(inp):
+    """Image Classifier Function.
+    Parameters
+    ----------
+    inp: Optional[np.ndarray] = None
+        Input image from callback
+    Returns
+    -------
+    Dict
+        A dictionary class names and its probability
+    """
+
+    # If input not valid, return dummy data or raise error
+    if inp is None:
+        return {'cat': 0.3, 'dog': 0.7}
+    
+    result = model(inp)
+
+    # postprocess
+    labeled_result = {class_names[label]: confidence for label, confidence in zip(result.top5, result.top5conf)}
+
+    return labeled_result
+
+# gradio code block for input and output
+with gr.Blocks() as app:
+    gr.Markdown("# Lung Cancer Classification")
+
+with open('index.html', encoding="utf-8") as f:
+    description = f.read()
+
+# gradio code block for input and output
+with gr.Blocks(theme=gr.themes.Default(primary_hue=page_utils.KALBE_THEME_COLOR, secondary_hue=page_utils.KALBE_THEME_COLOR).set(
+        button_primary_background_fill="*primary_600",
+        button_primary_background_fill_hover="*primary_500",
+        button_primary_text_color="white",
+    )) as app:
+    with gr.Column():
+        gr.HTML(description)
+
+    with gr.Row():
+        with gr.Column():
+            inp_img = gr.Image()
+            with gr.Row():
+                clear_btn = gr.Button(value="Clear")
+                process_btn = gr.Button(value="Process", variant="primary")
+        with gr.Column():
+            out_txt = gr.Label(label="Probabilities", num_top_classes=5)
+
+    process_btn.click(image_classifier, inputs=inp_img, outputs=out_txt)
+    clear_btn.click(lambda:(
+        gr.update(value=None),
+        gr.update(value=None)
+        ),
+        inputs=None,
+        outputs=[inp_img, out_txt])
+
+    gr.Markdown("## Image Examples")
+    gr.Examples(
+        examples=[os.path.join(examples_dir, "1.2.840.113564.1921681202.202011100756242032.1203801020003.dcm.jpeg")
+                 ],
+        inputs=inp_img,
+        outputs=out_txt,
+        fn=image_classifier,
+        cache_examples=False,
+    )
+    gr.Markdown(line_breaks=True, value='Author: Jason Adrian ([email protected]) <div class="row"><a href="https://github.com/jasonadriann?tab=repositories"><img alt="GitHub" src="https://img.shields.io/badge/Jason%20Adrian-000000?logo=github"> </div>')
+
+# demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
+app.launch(share=True)

requirements.txt

@@ -0,0 +1,69 @@
+asttokens==2.4.1
+certifi==2023.11.17
+charset-normalizer==3.3.2
+colorama==0.4.6
+comm==0.2.0
+contourpy==1.2.0
+cycler==0.12.1
+debugpy==1.8.0
+decorator==5.1.1
+exceptiongroup==1.2.0
+executing==2.0.1
+filelock==3.13.1
+fire==0.5.0
+fonttools==4.46.0
+fsspec==2023.12.1
+idna==3.6
+ipykernel==6.27.1
+ipython==8.18.1
+jedi==0.19.1
+Jinja2==3.1.2
+joblib==1.3.2
+jupyter_client==8.6.0
+jupyter_core==5.5.0
+kiwisolver==1.4.5
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+matplotlib-inline==0.1.6
+monai==1.3.0
+mpmath==1.3.0
+nest-asyncio==1.5.8
+networkx==3.2.1
+numpy==1.26.2
+opencv-python==4.8.1.78
+packaging==23.2
+pandas==2.1.3
+parso==0.8.3
+Pillow==10.1.0
+platformdirs==4.1.0
+prompt-toolkit==3.0.43
+psutil==5.9.6
+pure-eval==0.2.2
+py-cpuinfo==9.0.0
+Pygments==2.17.2
+pyparsing==3.1.1
+python-dateutil==2.8.2
+pytz==2023.3.post1
+pywin32==306
+PyYAML==6.0.1
+pyzmq==25.1.2
+requests==2.31.0
+scikit-learn==1.3.2
+scipy==1.11.4
+seaborn==0.13.0
+six==1.16.0
+stack-data==0.6.3
+sympy==1.12
+termcolor==2.4.0
+thop==0.1.1.post2209072238
+threadpoolctl==3.2.0
+torch==2.1.1
+torchvision==0.16.1
+tornado==6.4
+tqdm==4.66.1
+traitlets==5.14.0
+typing_extensions==4.8.0
+tzdata==2023.3
+ultralytics==8.0.225
+urllib3==2.1.0
+wcwidth==0.2.12

utils/page_utils.py

@@ -0,0 +1,51 @@
+from typing import Optional
+
+
+class ColorPalette:
+    """Color Palette Container."""
+    all = []
+
+    def __init__(
+        self,
+        c50: str,
+        c100: str,
+        c200: str,
+        c300: str,
+        c400: str,
+        c500: str,
+        c600: str,
+        c700: str,
+        c800: str,
+        c900: str,
+        c950: str,
+        name: Optional[str] = None,
+    ):
+        self.c50 = c50
+        self.c100 = c100
+        self.c200 = c200
+        self.c300 = c300
+        self.c400 = c400
+        self.c500 = c500
+        self.c600 = c600
+        self.c700 = c700
+        self.c800 = c800
+        self.c900 = c900
+        self.c950 = c950
+        self.name = name
+        ColorPalette.all.append(self)
+
+
+KALBE_THEME_COLOR = ColorPalette(
+    name='kalbe',
+    c50='#f2f9e8',
+    c100='#dff3c4',
+    c200='#c2e78d',
+    c300='#9fd862',
+    c400='#7fc93f',
+    c500='#3F831C',
+    c600='#31661a',
+    c700='#244c13',
+    c800='#18340c',
+    c900='#0c1b06',
+    c950='#050a02',
+)