Update app.py

app.py

@@ -4,7 +4,9 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import gradio as gr
+import os
 
+token = os.environ.get("HF_TOKEN")
 
 # Load default model as GPT2 and other models
 
@@ -16,7 +18,7 @@ tokenizer_gpt2 = AutoTokenizer.from_pretrained("openai-community/gpt2")
 model_gpt2 = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
 
 tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-2b")
-model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-2b")
+model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=token)
 
 tokenizer_Mistral = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
 model_Mistral = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")