Update app.py

app.py

@@ -2,15 +2,17 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import os
+load_dotenv()
 
+api_key = os.getenv("api_key")
 # App title and description
 st.title("I am Your GrowBuddy 🌱")
 st.write("Let me help you start gardening. Let's grow together!")
 
 def load_model():
     try:
-        tokenizer = AutoTokenizer.from_pretrained("KhunPop/Gardening")
-        model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it")
+        tokenizer = AutoTokenizer.from_pretrained("KhunPop/Gardening",  use_auth_token=api_key)
+        model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it",  use_auth_token=api_key)
         return tokenizer, model
     except Exception as e:
         st.error(f"Failed to load model: {e}")