Update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,5 @@
|
|
|
|
|
|
|
|
| 1 |
import matplotlib
|
| 2 |
matplotlib.use('Agg') # Non-GUI backend for plotting
|
| 3 |
|
|
@@ -22,7 +24,7 @@ os.makedirs(app.config["UPLOAD_FOLDER"], exist_ok=True)
|
|
| 22 |
# -----------------------------
|
| 23 |
# Load Keras classification model
|
| 24 |
try:
|
| 25 |
-
best_model = load_model("efficientnet_b0_best.
|
| 26 |
print("✅ EfficientNet model loaded successfully!")
|
| 27 |
except Exception as e:
|
| 28 |
print(f"❌ Error loading EfficientNet model: {e}")
|
|
|
|
| 1 |
+
|
| 2 |
+
|
| 3 |
import matplotlib
|
| 4 |
matplotlib.use('Agg') # Non-GUI backend for plotting
|
| 5 |
|
|
|
|
| 24 |
# -----------------------------
|
| 25 |
# Load Keras classification model
|
| 26 |
try:
|
| 27 |
+
best_model = load_model("efficientnet_b0_best.keras")
|
| 28 |
print("✅ EfficientNet model loaded successfully!")
|
| 29 |
except Exception as e:
|
| 30 |
print(f"❌ Error loading EfficientNet model: {e}")
|