anderni2 commited on
Commit
c3012c0
1 Parent(s): 138c721

Upload 4 files

Browse files
Files changed (5) hide show
  1. .gitattributes +1 -0
  2. README.md +25 -5
  3. Task_Pokemon.keras +3 -0
  4. app.py +32 -0
  5. 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: Pokemon
3
- emoji: 🏃
4
- colorFrom: gray
5
- colorTo: pink
6
  sdk: gradio
7
  sdk_version: 4.28.3
8
  app_file: app.py
9
  pinned: false
 
10
  ---
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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