---
title: Skin Lesion Segmentation Demo
emoji: 💻
colorFrom: pink
colorTo: purple
sdk: gradio
sdk_version: 5.31.0
app_file: app.py
pinned: false
license: mit
---

# 🧴 Skin Lesion Segmentation Demo

This interactive web application allows you to upload a skin image, select a segmentation model, and visualize the predicted lesion mask with optional post-processing.

## 🧠 Available Models
- **UNet**
- **SegFormer**
- **Inception-based CNN**
- **KMeans** (Unsupervised)
- **Gaussian Mixture Model** (Unsupervised)

## 🔧 Postprocessing Options
- None
- Morphological Open
- Morphological Close
- Erosion
- Dilation

## 💡 How it Works
The model takes in a skin lesion image (resized to 128×128), performs segmentation, and visualizes both:
- A binary predicted mask
- A color overlay on the original image

## 🛠️ Built With
- PyTorch & torchvision
- scikit-learn
- Hugging Face Transformers
- Gradio
- OpenCV

---

Demo created by Theodore Ioannidis (https://github.com/TheodoreIoannidis).