Update app.py
Browse files
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 |
-
|
| 12 |
-
from tensorflow.keras.
|
| 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:
|