Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ from datasets import load_dataset
|
|
| 5 |
from app.model import predict, gradcam, CLASS_NAMES
|
| 6 |
|
| 7 |
# Load HF dataset once at startup
|
| 8 |
-
dataset = load_dataset("AIOmarRehan/Brain_Tumor_MRI_Dataset")
|
| 9 |
|
| 10 |
# Convert any image to a usable PIL
|
| 11 |
def to_pil(example):
|
|
|
|
| 5 |
from app.model import predict, gradcam, CLASS_NAMES
|
| 6 |
|
| 7 |
# Load HF dataset once at startup
|
| 8 |
+
dataset = load_dataset("AIOmarRehan/Brain_Tumor_MRI_Dataset", split="train")
|
| 9 |
|
| 10 |
# Convert any image to a usable PIL
|
| 11 |
def to_pil(example):
|