second commit

app.py

@@ -102,8 +102,8 @@ def predict_on_frames(dir, mod_dir):
 	#for sorting the file names properly
 	files.sort(key = lambda x: int(x.stem[6:]))
 
-	for i in stqdm(range(len(files)), desc="Generating... this is a slow task"):
-		filename = Path(dir, files[i])
+	for file in stqdm(files, desc="Generating... this is a slow task"):
+		filename = Path(dir, file)
 		#print(filename)
 		#reading each files
 		img = Image.open(str(filename))
@@ -112,7 +112,7 @@ def predict_on_frames(dir, mod_dir):
 		pixel_values = img_ftr["pixel_values"].to(device)
 		# forward pass to get class logits and bounding boxes
 		outputs = model_pt(pixel_values=pixel_values)
-		mod_img_path = Path(mod_dir, files[i].name)
+		mod_img_path = Path(mod_dir, file.name)
 		save_predictions(img, outputs, mod_img_path)
 
 # Convert frames to video