aminahmed78 commited on
Commit
ccd9827
·
verified ·
1 Parent(s): 77f2142

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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")