Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from tensorflow.keras.applications.inception_v3 import preprocess_input
|
|
6 |
from tensorflow.keras.models import load_model
|
7 |
|
8 |
# Lade dein trainiertes Modell
|
9 |
-
model = load_model('/home/user/app/
|
10 |
|
11 |
def predict_character(img):
|
12 |
img = Image.fromarray(img.astype('uint8'), 'RGB') # Stelle sicher, dass das Bild im RGB-Format vorliegt
|
@@ -16,7 +16,7 @@ def predict_character(img):
|
|
16 |
img_array = preprocess_input(img_array) # Input für InceptionV3 vorverarbeiten
|
17 |
|
18 |
prediction = model.predict(img_array) # Vorhersage mit dem Modell
|
19 |
-
classes = ['
|
20 |
return {classes[i]: float(prediction[0][i]) for i in range(3)} # Vorhersage zurückgeben
|
21 |
|
22 |
# Definiere das Gradio-Interface
|
|
|
6 |
from tensorflow.keras.models import load_model
|
7 |
|
8 |
# Lade dein trainiertes Modell
|
9 |
+
model = load_model('/home/user/app/inceptionv3.h5') # Stelle sicher, dass dieser Pfad korrekt ist
|
10 |
|
11 |
def predict_character(img):
|
12 |
img = Image.fromarray(img.astype('uint8'), 'RGB') # Stelle sicher, dass das Bild im RGB-Format vorliegt
|
|
|
16 |
img_array = preprocess_input(img_array) # Input für InceptionV3 vorverarbeiten
|
17 |
|
18 |
prediction = model.predict(img_array) # Vorhersage mit dem Modell
|
19 |
+
classes = ['AttackonTitan', 'DragonBall', 'Tsubasa'] # Spezifische Charakter-Namen
|
20 |
return {classes[i]: float(prediction[0][i]) for i in range(3)} # Vorhersage zurückgeben
|
21 |
|
22 |
# Definiere das Gradio-Interface
|