Spaces:
Sleeping
Sleeping
Upload 4 files
Browse files- .gitattributes +1 -0
- README.md +25 -5
- Task_Pokemon.keras +3 -0
- app.py +32 -0
- requirements.txt +2 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
Task_Pokemon.keras filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
@@ -1,12 +1,32 @@
|
|
1 |
---
|
2 |
-
title:
|
3 |
-
emoji:
|
4 |
-
colorFrom:
|
5 |
-
colorTo:
|
6 |
sdk: gradio
|
7 |
sdk_version: 4.28.3
|
8 |
app_file: app.py
|
9 |
pinned: false
|
|
|
10 |
---
|
11 |
|
12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
title: Pokémon Klassifizierer
|
3 |
+
emoji: 🐢
|
4 |
+
colorFrom: pink
|
5 |
+
colorTo: green
|
6 |
sdk: gradio
|
7 |
sdk_version: 4.28.3
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
+
|
11 |
---
|
12 |
|
13 |
+
# Pokémon Klassifizierer
|
14 |
+
|
15 |
+
## Beschreibung
|
16 |
+
Dies ist eine einfache Machine Learning Anwendung, die Bilder von Pokémon klassifiziert. Lade ein Bild von einem der drei Pokémon Aerodactyl, Charizard oder Victreebel hoch, und die Anwendung wird dir mitteilen, um welches Pokémon es sich handelt und das Vertrauensniveau der Vorhersage angeben.
|
17 |
+
|
18 |
+
## Wie es funktioniert
|
19 |
+
- **Schritt 1**: Klicke auf "Durchsuchen" oder ziehe ein Bild in das Upload-Feld.
|
20 |
+
- **Schritt 2**: Das Bild wird automatisch verarbeitet, und das Modell gibt die Klasse und das Vertrauensniveau zurück.
|
21 |
+
|
22 |
+
## Modell
|
23 |
+
Das Modell wurde mit TensorFlow trainiert und kann die folgenden Pokémon erkennen:
|
24 |
+
- Aerodactyl
|
25 |
+
- Charizard
|
26 |
+
- Victreebel
|
27 |
+
|
28 |
+
## Technische Details
|
29 |
+
Dieser Space verwendet Gradio für die Benutzeroberfläche und TensorFlow für das Machine Learning Modell. Stelle sicher, dass alle Abhängigkeiten in der Datei `requirements.txt` aufgeführt sind.
|
30 |
+
|
31 |
+
## Über
|
32 |
+
Erstellt mit Leidenschaft von einem Pokémon-Liebhaber und Machine Learning Enthusiasten. Genieße die Nutzung!
|
Task_Pokemon.keras
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a798fcc70a364958ea084cf69eea60f837c173c3a3da66c02679839a687da791
|
3 |
+
size 283472566
|
app.py
ADDED
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import tensorflow as tf
|
3 |
+
from PIL import Image
|
4 |
+
import numpy as np
|
5 |
+
|
6 |
+
# Lade dein benutzerdefiniertes Regressionsmodell
|
7 |
+
model = tf.keras.models.load_model('Task_Pokemon.keras')
|
8 |
+
|
9 |
+
# Klassennamen, sollten deinem Dataset entsprechen
|
10 |
+
class_names = ['Aerodactyl', 'Charizard', 'Victreebel']
|
11 |
+
|
12 |
+
def classify_image(image):
|
13 |
+
img = image.resize((160, 160)) # Hier definieren wir die Größe der Bilder
|
14 |
+
img_array = tf.keras.preprocessing.image.img_to_array(img)
|
15 |
+
img_array = tf.expand_dims(img_array, 0) # Erstelle einen Batch
|
16 |
+
predictions = model.predict(img_array)
|
17 |
+
predicted_class = class_names[np.argmax(predictions[0])]
|
18 |
+
confidence = np.max(predictions[0])
|
19 |
+
return predicted_class, confidence
|
20 |
+
|
21 |
+
image_input = gr.inputs.Image()
|
22 |
+
label = gr.outputs.Label(num_top_classes=3)
|
23 |
+
|
24 |
+
iface = gr.Interface(
|
25 |
+
fn=classify_image,
|
26 |
+
inputs=image_input,
|
27 |
+
outputs=label,
|
28 |
+
title='Pokémon Klassifizierer',
|
29 |
+
description='Lade ein Bild von Aerodactyl, Charizard oder Victreebel hoch und der Klassifizierer wird dir mitteilen, um welches Pokémon es sich handelt, sowie das Vertrauensniveau der Vorhersage.'
|
30 |
+
)
|
31 |
+
|
32 |
+
iface.launch()
|
requirements.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
gradio
|
2 |
+
tensorflow
|