UdaraChamidu commited on
Commit
c94086d
·
verified ·
1 Parent(s): 1e55d39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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.h5")
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}")