ademibeh commited on
Commit
52a75c7
1 Parent(s): a21c3cc

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('/Users/beharademi/Documents/ZHAW/6. Sem LOKAL/KI/module exam/ademibeh/resnet_best_model.keras') # Update path
11
- mobilenet_model = load_model('/Users/beharademi/Documents/ZHAW/6. Sem LOKAL/KI/module exam/ademibeh/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')