ademibeh commited on
Commit
630059d
1 Parent(s): 510a562

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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('/path/to/resnet_best_model.keras') # Update path
11
- mobilenet_model = load_model('/path/to/mobilenet_best_model.keras') # Update path
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')