mandanah commited on
Commit
706adef
·
1 Parent(s): 9351f4a

corrected checkpoint path

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import tensorflow as tf
4
  import matplotlib.pyplot as plt
5
  import numpy as np
6
  from PIL import Image
7
- from turtle import width
8
  import streamlit as st
9
 
10
 
@@ -43,7 +43,8 @@ feature_extraction_model = 'ResNet152'
43
  tokenizer_path = 'tokenizer.pkl'
44
  # checkpoint_path = "/dbfs/FileStore/shared_uploads/mhajiza@gap.com/computer_vision/models/image_captioning_tf_14/ckpt-10"
45
  # checkpoint_path = "/dbfs/FileStore/shared_uploads/mhajiza@gap.com/computer_vision/models/image_captioning_tf_14/manually_saved_model-11"
46
- checkpoint_path = "/Users/mhajiza/Documents/Computer_Vison/Image_captioning/image_captioning_tf_model/ckpt-10"
 
47
  # checkpoint_path = "/Users/mhajiza/Documents/Computer_Vison/Image_captioning/image_captioning_tf_model/manually_saved_model-11"
48
 
49
  # COMMAND ----------
 
4
  import matplotlib.pyplot as plt
5
  import numpy as np
6
  from PIL import Image
7
+ #from turtle import width
8
  import streamlit as st
9
 
10
 
 
43
  tokenizer_path = 'tokenizer.pkl'
44
  # checkpoint_path = "/dbfs/FileStore/shared_uploads/mhajiza@gap.com/computer_vision/models/image_captioning_tf_14/ckpt-10"
45
  # checkpoint_path = "/dbfs/FileStore/shared_uploads/mhajiza@gap.com/computer_vision/models/image_captioning_tf_14/manually_saved_model-11"
46
+ # checkpoint_path = "/Users/mhajiza/Documents/Computer_Vison/Image_captioning/image_captioning_tf_model/ckpt-10"
47
+ checkpoint_path = "ckpt-10"
48
  # checkpoint_path = "/Users/mhajiza/Documents/Computer_Vison/Image_captioning/image_captioning_tf_model/manually_saved_model-11"
49
 
50
  # COMMAND ----------