Update app.py

app.py

@@ -1,10 +1,11 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# Ensure you're logged in
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b-it", use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained("google/gemma-7b-it", use_auth_token=True)
 
-# Load the tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b-it")
-model = AutoModelForCausalLM.from_pretrained("google/gemma-7b-it", torch_dtype=torch.bfloat16)
 
 # Set up the Streamlit page configuration
 st.set_page_config(page_title="AI Companion Chatbot", layout="centered")