Update app.py

app.py

@@ -8,6 +8,13 @@ from huggingface_hub import HfApi
 import pandas as pd
 #from transformers import pipeline
 
+
+import torch
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
+
 load_dotenv()
 
 HF_TOKEN = os.getenv("HF_TOKEN")