prepare captions.csv

app.py

@@ -31,17 +31,18 @@ cmd = "ls -l"
 output1 = subprocess.check_output(cmd, shell=True).decode("utf-8")
 print(output1)
 
-cmd = "ls Images"
-output1 = subprocess.check_output(cmd, shell=True).decode("utf-8")
-print(output1)
-
 cmd = "pwd"
 output1 = subprocess.check_output(cmd, shell=True).decode("utf-8")
-print(output1)
+print(output1) # result => /home/user/app
 #=============================================
 print("Going to invoke make_train_valid_dfs")
 print("\n\n")
 
+df = pd.read_csv("captions.txt")
+df['id'] = [id_ for id_ in range(df.shape[0] // 5) for _ in range(5)]
+df.to_csv("captions.csv", index=False)
+df = pd.read_csv("captions.csv")
+
 _, valid_df = make_train_valid_dfs()
 model, image_embeddings = get_image_embeddings(valid_df, "best.pt")