Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,8 +7,8 @@ from tensorflow.keras.applications.mobilenet_v2 import preprocess_input as mobil
|
|
7 |
from tensorflow.keras.models import load_model
|
8 |
|
9 |
# Load your trained models
|
10 |
-
resnet_model = load_model('/
|
11 |
-
mobilenet_model = load_model('/
|
12 |
|
13 |
def predict_comic_character(img, model_type):
|
14 |
img = Image.fromarray(img.astype('uint8'), 'RGB')
|
|
|
7 |
from tensorflow.keras.models import load_model
|
8 |
|
9 |
# Load your trained models
|
10 |
+
resnet_model = load_model('/home/user/app/resnet_best_model.keras') # Update path
|
11 |
+
mobilenet_model = load_model('/home/user/app/mobilenet_best_model.keras') # Update path
|
12 |
|
13 |
def predict_comic_character(img, model_type):
|
14 |
img = Image.fromarray(img.astype('uint8'), 'RGB')
|