Update app.py

app.py

@@ -2,6 +2,12 @@ import gradio as gr
 import os
 from huggingface_hub import login
 
+
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
+
 # # api_key = os.getenv('llama3token')
 # # login(api_key)