Hugging Face's logo

hoangthan
/
img-caption-demo

Image-to-Text
Keras
TF-Keras
Model card Files Community
1
hoangthan commited on
Commit
766f2b3
•
1 Parent(s): d1f0eea

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -7,7 +7,7 @@ import pickle
7
  from PIL import Image
8
 
9
  class PreTrainedPipeline():
10
- def __init__(self, path=""):
11
 
12
  self.model = keras.models.load_model(os.path.join(path, ""))
13
 
 
7
  from PIL import Image
8
 
9
  class PreTrainedPipeline():
10
+ def __init__(self, path="model"):
11
 
12
  self.model = keras.models.load_model(os.path.join(path, ""))
13