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