Upload 3 files

app.py

@@ -0,0 +1,30 @@
+import gradio as gr
+from PIL import Image
+import numpy as np
+from tensorflow.keras.preprocessing import image as keras_image
+from tensorflow.keras.applications.resnet50 import preprocess_input
+from tensorflow.keras.models import load_model
+
+# Load your trained model
+model = load_model('C:\Users\Kevin Keo\OneDrive - ZHAW\Dokumente\Desktop\ZHAW WIN\FS24\KI-Anwendung\Pokemon\Mandatory Sauber\Mandatory Sauber\mein_modell.h5')  # Ensure this path is correct
+
+def predict_pokemon(img):
+    img = Image.fromarray(img.astype('uint8'), 'RGB')  # Ensure the image is in RGB
+    img = img.resize((224, 224))  # Resize the image properly using PIL
+    img_array = keras_image.img_to_array(img)  # Convert the image to an array
+    img_array = np.expand_dims(img_array, axis=0)  # Expand dimensions to fit model input
+    img_array = preprocess_input(img_array)  # Preprocess the input as expected by ResNet50
+    
+    prediction = model.predict(img_array)  # Predict using the model
+    classes = ['Butterfree', 'Cubone', 'Dratini' ]  # Specific Pokémon names
+    return {classes[i]: float(prediction[0][i]) for i in range(3)}  # Return the prediction
+
+# Define Gradio interface
+interface = gr.Interface(fn=predict_pokemon, 
+                         inputs="image",  # Simplified input type
+                         outputs="label",  # Simplified output type
+                         title="Pokémon Classifier",
+                         description="Upload an image of a Pokémon and the classifier will predict its species.")
+
+# Launch the interface
+interface.launch()

mein_modell.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f95d83c6b5aa515070432d124f46c6c7e8c13f9c1e86a9b87c221d5872b59df
+size 308621824

requirements.txt

@@ -0,0 +1,19 @@
+blinker==1.7.0
+click==8.1.7
+Flask==3.0.2
+Flask-Cors==4.0.0
+itsdangerous==2.1.2
+Jinja2==3.1.3
+joblib==1.3.2
+MarkupSafe==2.1.5
+numpy==1.26.4
+pandas==2.2.1
+python-dateutil==2.8.2
+pytz==2024.1
+scikit-learn==1.4.1.post1
+scipy==1.12.0
+six==1.16.0
+threadpoolctl==3.3.0
+tzdata==2024.1
+Werkzeug==3.0.1
+tensorflow==2.16.1