Update app.py

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 from transformers import pipeline, BitsAndBytesConfig, AutoModelForCausalLM, AutoTokenizer
 import torch 
 
+
+print(torch.cuda.is_available())
+
 # Load the Hugging Face model for chatbot
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,