Update app.py

app.py

@@ -18,6 +18,7 @@ def init_model(model_id, is_pf=False):
         path = hf_hub_download(repo_id="jameslahm/yoloe", filename=f"{model_id}-seg-pf.pt")
         model = YOLOE(path)
     model.eval()
+    model.to("cuda")
     return model
 
 @spaces.GPU