kai-sheng commited on
Commit
9ba935a
·
verified ·
1 Parent(s): ded1a34

Update model name

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -84,7 +84,7 @@ def generate_caption():
84
  result = format_tesseract_output(extracted_text)
85
  else:
86
  tokenizer = load(open("tokenizer.p","rb"))
87
- model = load_model('model_9.keras')
88
 
89
  xception_model = Xception(include_top=False, pooling="avg")
90
  photo = extract_features(image_data, xception_model)
 
84
  result = format_tesseract_output(extracted_text)
85
  else:
86
  tokenizer = load(open("tokenizer.p","rb"))
87
+ model = load_model('model.keras')
88
 
89
  xception_model = Xception(include_top=False, pooling="avg")
90
  photo = extract_features(image_data, xception_model)