Update app.py

app.py

@@ -3,14 +3,13 @@ from PIL import Image
 import numpy as np
 import gradio as gr
 from tensorflow import keras
-
-# Load the model (replace with the correct path on Hugging Face Spaces)
-import tensorflow as tf
+from keras.models import load_model
 
 model_path = "VGG_16.h5"
-input_shape = (224, 224, 3)  # Replace with your input shape
-model = tf.keras.models.load_model(model_path, compile=False)
-model.build(input_shape)
+model = load_model(model_path, compile=False)
+
+
+
 
 # Define the class labels
 class_labels = ['biotite', 'granite', 'olivine', 'plagioclase', 'staurolite']