Update app.py

app.py

@@ -2,10 +2,12 @@ import gradio as gr
 from ultralyticsplus import YOLO, render_result
 from ultralytics.yolo.utils.plotting import Annotator
 
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model = YOLO('eeshawn11/naruto_hand_seal_detection')
 model.conf = 0.50
 model.iou = 0.45
 model.overrides['max_det'] = 1
+model.to(device)
 
 def yolov8_inference(
     image: gr.Image = None,