Update app.py

app.py

@@ -48,7 +48,6 @@ if 'confidence' not in st.session_state:
     st.session_state.confidence = 0.50
 
 if 'model' not in st.session_state:
-    torch.cuda.set_device(0)
     st.session_state.model = YOLO("/home/user/app/models/head_model/best.pt")  # load 
 
 _, row0_col1, _ = st.columns([2,3,2])