khurrameycon commited on
Commit
d490720
·
verified ·
1 Parent(s): 246525b

try for T4

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ from PyPDF2 import PdfReader
10
  from threading import Thread
11
 
12
  # Check if we're running in a Hugging Face Space and if SPACES_ZERO_GPU is enabled
13
- IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
14
- IS_SPACE = os.environ.get("SPACE_ID", None) is not None
15
 
16
  # Determine the device (GPU if available, else CPU)
17
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
10
  from threading import Thread
11
 
12
  # Check if we're running in a Hugging Face Space and if SPACES_ZERO_GPU is enabled
13
+ # IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
14
+ # IS_SPACE = os.environ.get("SPACE_ID", None) is not None
15
 
16
  # Determine the device (GPU if available, else CPU)
17
  device = "cuda" if torch.cuda.is_available() else "cpu"