Update app.py

app.py

@@ -14,13 +14,13 @@ from PIL import Image
 import gradio as gr
 
 # Load your trained model
-model = tf.keras.models.load_model('/content/model.h5', custom_objects={'KerasLayer': hub.KerasLayer})
+model = tf.keras.models.load_model('model.h5', custom_objects={'KerasLayer': hub.KerasLayer})
 
 # Define the image size
 IMG_SIZE = 224
 
 # Load class names from the text file
-with open('/content/class_names.txt', 'r') as file:
+with open('class_names.txt', 'r') as file:
     class_names = [line.strip() for line in file]
 
 # Define a function to preprocess the image