Update app.py

app.py

@@ -21,6 +21,13 @@ def predict(filename, model=None, device=None, reduce_labels=True):
     
     global default_model
     assert isinstance(filename, str)
+
+    if device is None:
+        if torch.cuda.device_count():
+            device = 'cuda'
+        else:
+            device = 'cpu'
+    
     print(dict(
         filename=filename,
         model=model,