Update app.py

app.py

@@ -4,10 +4,10 @@ from PIL import Image
 import numpy as np
 
 # Lade dein Modell (hier als Beispiel die Keras .h5 Datei)
-model = tf.keras.models.load_model('pokemon_model.keras')
+model = tf.keras.models.load_model('gym_equipment_transferlearning.keras')
 
 # Klassennamen, sollten deinem Dataset entsprechen
-class_names = ['Jolteon', 'Kakuna', 'Mr. Mime']
+class_names = ['benchPress', 'dumbBell', 'kettleBell', 'treadMill']
 
 def classify_image(image):
     image = Image.fromarray(image.astype('uint8'), 'RGB')
@@ -27,7 +27,7 @@ iface = gr.Interface(
     fn=classify_image, 
     inputs=image_input, 
     outputs=label,
-    title='Pokémon Classifier',
+    title='Gym Equipment Classifier',
              description='Upload an image of Jolteon, Kakuna, Mr. Mime and the classifier will tell you which one it is and the confidence level of the prediction.').launch()
 
 iface.launch()