To cpu

app.py

@@ -107,6 +107,10 @@ def detect_objects(
         )
 
         results.extend(batch_results)
+
+    # move results to cpu
+    for i, result in enumerate(results):
+        results[i] = {k: v.cpu() for k, v in result.items()}
     
     return results, model.config.id2label