raja5259 commited on
Commit
8fa17f6
·
verified ·
1 Parent(s): 13f3e99

find captions.csv in two dir only

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ data_source = 'flickr8k.zip'
16
 
17
  with zipfile.ZipFile(data_source, 'r') as zip_ref:
18
  zip_ref.extractall('.')
19
- os.system("find / -name captions.csv")
20
-
21
 
22
  _, valid_df = make_train_valid_dfs()
23
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")
 
16
 
17
  with zipfile.ZipFile(data_source, 'r') as zip_ref:
18
  zip_ref.extractall('.')
19
+ os.system("find /home/user/ -name captions.csv")
20
+ os.system("find /usr/ -name captions.csv")
21
 
22
  _, valid_df = make_train_valid_dfs()
23
  model, image_embeddings = get_image_embeddings(valid_df, "best.pt")