Habith commited on
Commit
e75df92
·
verified ·
1 Parent(s): be28309

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +49 -0
README.md ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: GoGenix MRI Brain Tumor Classifier
3
+ emoji: ⚡
4
+ colorFrom: blue
5
+ colorTo: green
6
+ sdk: gradio
7
+ sdk_version: 4.0.0
8
+ python_version: '3.10'
9
+ app_file: app.py
10
+ pinned: true
11
+ license: mit
12
+ short_description: quick-predictions
13
+ ---
14
+
15
+ # GoGenix MRI Brain Tumor Classifier
16
+
17
+ Fine-tuned version of `Falconsai/nsfw_image_detection` for Brain Tumor MRI classification.
18
+
19
+ ## 🚀 Model Information
20
+ - **Base Model**: [Falconsai/nsfw_image_detection](https://huggingface.co/Falconsai/nsfw_image_detection)
21
+ - **Custom Model**: GoGenix_MRI_Brain
22
+ - **Number of Classes**: 4 (Glioma, Meningioma, No Tumor, Pituitary Tumor)
23
+ - **Dataset**: Brain Tumor MRI 4-class Dataset
24
+
25
+ ## 📊 Dataset Information
26
+ The model is trained on a comprehensive Brain Tumor MRI dataset containing:
27
+ - **Glioma Tumors**
28
+ - **Meningioma Tumors**
29
+ - **Pituitary Tumors**
30
+ - **No Tumor** (Healthy brains)
31
+
32
+ ## 🎯 Usage
33
+ ### Training
34
+ 1. Navigate to the **"🚀 Train Model"** tab
35
+ 2. Click **"Start Training"** to begin fine-tuning
36
+ 3. Monitor the training progress in the status box
37
+
38
+ ### Classification
39
+ 1. Go to the **"🔍 Classify MRI"** tab
40
+ 2. Upload a Brain MRI image
41
+ 3. Click **"Classify"** to get tumor classification results
42
+
43
+ ## 🔧 Technical Details
44
+ - **Framework**: PyTorch + Hugging Face Transformers
45
+ - **Image Size**: 224x224 pixels
46
+ - **Architecture**: Based on Vision Transformer (ViT)
47
+ - **Training Time**: ~10-30 minutes on T4 GPU
48
+
49
+ ## 📁 Files Structure