Update app.py

app.py

@@ -1,15 +1,17 @@
-import gradio as gr
+import os
 import tensorflow as tf
-from PIL import Image
-import numpy as np
+
+# Disable all GPUS
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
 # Load your pre-trained model
 def load_model():
-    model = tf.keras.models.load_model('depi-graduation-project.ipynb')  # Replace with your model's path
+    model = tf.keras.models.load_model('satellite_model.h5')  # Replace with your model's path
     return model
 
 model = load_model()
 
+
 # Define the labels (categories)
 labels = ['Water', 'Cloudy', 'Desert', 'Green Area']