Hugging Face's logo

Spaces:
shredder-31
/
ImgCap
Sleeping

App Files Community
shredder-31 commited on
Commit
a0a02a7
·
verified ·
1 Parent(s): d7027f7

Update trainning.py

Browse files
Files changed (1) hide show
  1. trainning.py +1 -1
trainning.py CHANGED
@@ -7,7 +7,7 @@ import torchvision.models as models
7
 
8
  def decoder(indices):
9
 
10
- with open(f"/teamspace/studios/this_studio/ImgCap/vocab.pkl", 'rb') as f:
11
  vocab = pickle.load(f)
12
 
13
  tokens = [vocab.lookup_token(idx) for idx in indices]
 
7
 
8
  def decoder(indices):
9
 
10
+ with open("vocab.pkl", 'rb') as f:
11
  vocab = pickle.load(f)
12
 
13
  tokens = [vocab.lookup_token(idx) for idx in indices]