NikhilPatil commited on
Commit
78dcac1
·
verified ·
1 Parent(s): 0bf841c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -7,9 +7,9 @@ import pickle
7
  import torch
8
  import numpy as np
9
  from PIL import Image
10
- from keras.models import load_model
11
- #from tensorflow.keras.models import load_model
12
- from tensorflow.keras.preprocessing.image import img_to_array, load_img
13
  from transformers import CLIPProcessor, CLIPModel
14
  from sentence_transformers import SentenceTransformer, CrossEncoder
15
  from langdetect import detect
@@ -20,11 +20,10 @@ model_path = os.path.join(Main_py, "best_cnn_model_finetuned.keras")
20
  label_path = os.path.join(Main_py, "label_encoder.pkl")
21
  json_path = os.path.join(Main_py, "banana_disease_knowledge_base_updated_shuffled.json")
22
 
23
-
24
  # ===================== LOAD MODELS & DATA =====================
25
  @st.cache_resource
26
  def load_cnn_clip_kb():
27
- model = load_model(model_path)
28
  with open(label_path, "rb") as f:
29
  le = pickle.load(f)
30
  with open(json_path, "r", encoding="utf-8") as f:
 
7
  import torch
8
  import numpy as np
9
  from PIL import Image
10
+ from tensorflow.keras.models import load_model # ✅ FIXED
11
+ from tensorflow.keras.preprocessing.image import img_to_array, load_img # ✅ FIXED
12
+ from tensorflow.keras.callbacks import EarlyStopping, ModelCheckpoint # optional, if used
13
  from transformers import CLIPProcessor, CLIPModel
14
  from sentence_transformers import SentenceTransformer, CrossEncoder
15
  from langdetect import detect
 
20
  label_path = os.path.join(Main_py, "label_encoder.pkl")
21
  json_path = os.path.join(Main_py, "banana_disease_knowledge_base_updated_shuffled.json")
22
 
 
23
  # ===================== LOAD MODELS & DATA =====================
24
  @st.cache_resource
25
  def load_cnn_clip_kb():
26
+ model = load_model(model_path) # ✅ FIXED
27
  with open(label_path, "rb") as f:
28
  le = pickle.load(f)
29
  with open(json_path, "r", encoding="utf-8") as f: