EJ-L commited on
Commit
dbc0e14
·
1 Parent(s): fc76c30

change cpu to gpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ import re
26
 
27
  # -------- Runtime / device --------
28
  # Force CPU usage
29
- device = "cpu"
30
 
31
  # Hugging Face Spaces port
32
  PORT = int(os.getenv("PORT", "7860"))
 
26
 
27
  # -------- Runtime / device --------
28
  # Force CPU usage
29
+ device = "cuda"
30
 
31
  # Hugging Face Spaces port
32
  PORT = int(os.getenv("PORT", "7860"))