final

app.py

@@ -56,12 +56,14 @@ def predict_video(video_path):
             break
         
         frame_count += 1
+        print(frame_count)
         image = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
         image = transform(image).unsqueeze(0).to(device)
 
         with torch.no_grad():
             outputs = model(image)
             _, predicted = torch.max(outputs, 1)
+        print(predicted)
         
         if predicted.item() == 0:
             real_count += 1