Update app.py

app.py

@@ -59,7 +59,6 @@ def inference(device, model_type, points_per_side, pred_iou_thresh, stability_sc
         frames_num = cap.get(cv2.CAP_PROP_FRAME_COUNT)
         W, H = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
         fps = int(cap.get(cv2.CAP_PROP_FPS))
-        print(fps)
         out = cv2.VideoWriter("output.mp4", cv2.VideoWriter_fourcc('x', '2', '6', '4'), fps, (W, H), isColor=True)
         for _ in progress.tqdm(range(int(frames_num)), desc='Processing video ({} frames, size {}x{})'.format(int(frames_num), W, H)):
             ret, frame = cap.read()     # read a frame
@@ -83,7 +82,7 @@ with gr.Blocks() as demo:
             # select model
             model_type = gr.Dropdown(["vit_b", "vit_l", "vit_h"], value='vit_b', label="Select Model")
             # select device
-            device = gr.Dropdown(["cpu", "cuda"], value='cuda', label="Select Device")
+            device = gr.Dropdown(["cpu"], value='cpu', label="Select Device")
 
     # 参数
     with gr.Accordion(label='Parameters', open=False):