Update app.py

app.py

@@ -5,7 +5,9 @@ from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 
 # Load model and tokenizer (Gemma 2B or similar)
-model_id = "google/gemma-2b-it"
+model_id = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+login(token=os.environ["HUGGINGFACE_API_TOKEN"])
+
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype="auto")