Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,9 +58,9 @@ async def load_model():
|
|
| 58 |
# Set cache directory if not already set
|
| 59 |
import os
|
| 60 |
if not os.environ.get("TRANSFORMERS_CACHE"):
|
| 61 |
-
os.environ["TRANSFORMERS_CACHE"] = "/
|
| 62 |
if not os.environ.get("HF_HOME"):
|
| 63 |
-
os.environ["HF_HOME"] = "/
|
| 64 |
|
| 65 |
# Determine device
|
| 66 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 58 |
# Set cache directory if not already set
|
| 59 |
import os
|
| 60 |
if not os.environ.get("TRANSFORMERS_CACHE"):
|
| 61 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp"
|
| 62 |
if not os.environ.get("HF_HOME"):
|
| 63 |
+
os.environ["HF_HOME"] = "/tmp"
|
| 64 |
|
| 65 |
# Determine device
|
| 66 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|