Dominik's gradio

README.md

@@ -1,13 +1,12 @@
 ---
-title: Odontathon Chest Xray
-emoji: 📈
-colorFrom: pink
-colorTo: blue
+title: Pneumonia probability - chest x-ray
+emoji: 🧙
+colorFrom: indigo
+colorTo: gray
 sdk: gradio
 sdk_version: 3.5
-app_file: app.py
+python_version: 3.10.5
+app_file: application.py
 pinned: false
-license: mit
+tags: [odontathon, chest-x-ray, pneumonia, probability]
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

application.py

@@ -0,0 +1,21 @@
+import gradio
+from transformers import pipeline
+
+"""Predicts the probability of pneumonia given a chest x-ray image."""
+
+
+_pipeline = pipeline(task="image-classification", model="nickmuchi/vit-finetuned-chest-xray-pneumonia")
+
+def predict(image):
+  predictions = _pipeline(image)
+  return {p["label"]: p["score"] for p in predictions}
+
+
+demo = gradio.Interface(
+    fn=predict,
+    inputs=gradio.inputs.Image(label="Upload lung x-ray image", type="filepath"),
+    outputs=gradio.outputs.Label(num_top_classes=2),
+    title="Pneumonia Probability",
+)
+
+demo.launch()

requirements.txt

@@ -0,0 +1,7 @@
+transformers
+gradio
+Jinja2=="3.0.3"
+aiofiles
+torch
+torchvision
+torchaudio