Update app.py

app.py

@@ -98,9 +98,9 @@ def create_vector_store(transcript):
 # -------------------------------------------------
 def build_model(model_choice, temperature=0.7):
     """Return the correct model and a flag indicating if it’s chat-based."""
-    if model_choice == "TinyLlama-1.1B":
+    if model_choice == "Llama-3.2-1B":
         llm = HuggingFaceEndpoint(
-            repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
+            repo_id="meta-llama/Llama-3.2-1B-Instruct",
             huggingfacehub_api_token=api_key,
             task="text-generation",
             max_new_tokens=500,
@@ -162,7 +162,7 @@ st.title("🎬 YouTube Transcript Chatbot (RAG)")
 
 video_id = st.text_input("YouTube Video ID", value="lv1_-RER4_I")
 query = st.text_area("Your Query", value="What is RAG?")
-model_choice = st.radio("Model to Use", ["TinyLlama-1.1B", "Gemma-2-3B", "DeepSeek-685B", "OpenAI-20B"])
+model_choice = st.radio("Model to Use", ["Llama-3.2-1B", "Gemma-2-3B", "DeepSeek-685B", "OpenAI-20B"])
 temperature = st.slider("Temperature", 0, 100, value=50) / 100.0
 
 # Get available languages for this video