Update app.py

app.py

@@ -1,3 +1,4 @@
+import torch
 import numpy as np
 import streamlit as st
 from openai import OpenAI
@@ -6,6 +7,10 @@ import sys
 from dotenv import load_dotenv, dotenv_values
 load_dotenv()
 
+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
 
 # initialize the client
 client = OpenAI(