franckew commited on
Commit
194d92f
1 Parent(s): 4c7f0e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/one_piece_character_classifier.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,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 = ['Brook', 'Chopper', 'Zoro'] # 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
 
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