Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ import io
|
|
| 13 |
INPUT_SIZE = 224
|
| 14 |
|
| 15 |
# Chemins des modèles sauvegardés (adapté pour PyTorch)
|
| 16 |
-
MODEL_PYTORCH_PATH = "
|
| 17 |
|
| 18 |
# Chargement des classes
|
| 19 |
class_names = ['Healthy', 'Tumor']
|
|
|
|
| 13 |
INPUT_SIZE = 224
|
| 14 |
|
| 15 |
# Chemins des modèles sauvegardés (adapté pour PyTorch)
|
| 16 |
+
MODEL_PYTORCH_PATH = "VGG16_best_accuracy_0.9660.pth" # Exemple
|
| 17 |
|
| 18 |
# Chargement des classes
|
| 19 |
class_names = ['Healthy', 'Tumor']
|