Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from tensorflow.keras.applications.resnet50 import preprocess_input
|
|
6 |
from tensorflow.keras.models import load_model
|
7 |
|
8 |
# Load your trained model
|
9 |
-
model = load_model('/
|
10 |
|
11 |
def predict_pokemon(img):
|
12 |
img = Image.fromarray(img.astype('uint8'), 'RGB') # Ensure the image is in RGB
|
|
|
6 |
from tensorflow.keras.models import load_model
|
7 |
|
8 |
# Load your trained model
|
9 |
+
model = load_model('/home/user/app/mein_modell.h5')
|
10 |
|
11 |
def predict_pokemon(img):
|
12 |
img = Image.fromarray(img.astype('uint8'), 'RGB') # Ensure the image is in RGB
|