Update app.py

app.py

@@ -1,6 +1,6 @@
 from io import BytesIO
 import base64
-
+import argparse
 import numpy as np
 import torch
 import torch.nn as nn
@@ -657,8 +657,10 @@ def capture_frame_from_webcam(duration=7):
         transforms.ToTensor(),
         transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
     ])
-
-    cap = cv2.VideoCapture(0)  # Open the default webcam (usually ID 0)
+    args = parse_args()
+    device = torch.device(args.device)
+    
+    cap = cv2.VideoCapture(args.camera_id)  # Open the default webcam (usually ID 0)
 
     frames = []
     start_time = time.time()