Update app.py

app.py

@@ -9,18 +9,16 @@ app = FastAPI()
 
 # Load your model
 import os
-import tensorflow as tf
+
 
 model_path = "models/2.keras"
 
-# Just check if file is accessible
-if os.path.exists(model_path):
-    model = tf.keras.models.load_model(model_path)
-else:
-    raise FileNotFoundError("Model not found at: models/2.keras")
+with tf.io.gfile.GFile(model_path, 'rb') as f:
+    model = tf.keras.models.load_model(f)
+
 
 # Now load the model
-model = tf.keras.models.load_model(model_path)
+
 
 CLASS_NAMES = ['Fungi', 'Healthy', 'Nematode', 'Pest', 'Phytopthora', 'Virus']