Update app.py

app.py

@@ -2,9 +2,11 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
 
-# 🔹 Load Llama 3 - 7B model
-model_name = "meta-llama/Meta-Llama-3-7B-Instruct"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+# ✅ Use Llama 2 - 7B Chat (Public)
+model_name = "meta-llama/Llama-2-7b-chat-hf"
+
+# 🔹 Authenticate if required (for gated models)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
 
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=100)
@@ -41,7 +43,7 @@ demo = gr.Interface(
     fn=recommend_gifts, 
     inputs="text", 
     outputs="json",
-    title="🎁 AI Gift Recommender (Llama 3 - 7B)",
+    title="🎁 AI Gift Recommender (Llama 2 - 7B)",
     description="Enter details about the person and get personalized gift suggestions with shopping links!",
 )