Spaces:
Sleeping
Sleeping
try for T4
Browse files
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"
|