Update app.py

app.py

@@ -1,3 +1,4 @@
+device = torch.device("cpu")
 import subprocess
 import sys
 import gradio as gr