File size: 702 Bytes
5073ec7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
---

title: GERD Lightweight ViT Classifier
emoji: 🏥
colorFrom: blue
colorTo: green
sdk: gradio
sdk_version: 4.0.0
app_file: app.py
pinned: false
---


# GERD Lightweight Vision Transformer Classifier

This is a lightweight Vision Transformer model for GERD classification.

## Model Architecture

- **Image size**: 224×224×3
- **Patch size**: 8×8
- **Projection dimension**: 64
- **Transformer layers**: 4
- **Attention heads**: 4
- **MLP head units**: [128, 64]

## Usage

Upload an image and the model will classify it into one of the predefined classes.

## Preprocessing

Images are:

1. Resized to 224×224
2. Converted to RGB
3. Normalized to [0, 1] range