Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,9 @@ import os
|
|
3 |
from transformers import pipeline
|
4 |
from langdetect import detect
|
5 |
from groq import Groq
|
|
|
|
|
|
|
6 |
|
7 |
# Load Hugging Face token from environment
|
8 |
HF_TOKEN = os.environ.get("homeo_doc")
|
|
|
3 |
from transformers import pipeline
|
4 |
from langdetect import detect
|
5 |
from groq import Groq
|
6 |
+
import torch
|
7 |
+
print(torch.__version__)
|
8 |
+
print("CUDA available:", torch.cuda.is_available()) # Check if GPU is available
|
9 |
|
10 |
# Load Hugging Face token from environment
|
11 |
HF_TOKEN = os.environ.get("homeo_doc")
|