Update app.py

app.py

@@ -8,7 +8,7 @@ from timm.data import create_transform
 from typing import Any
 
 
-device = "cuda"
+device = "cpu"
 
 # Prepare the model.
 import models