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 == "Flan-T5 (Free)":
+    if model_choice == "TinyLlama-1.1B":
         llm = HuggingFaceEndpoint(
-            repo_id="google/flan-t5-base",
+            repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
             huggingfacehub_api_token=api_key,
             task="text-generation",
             max_new_tokens=500,
@@ -108,7 +108,16 @@ def build_model(model_choice, temperature=0.7):
         )
         return ChatHuggingFace(llm=llm, temperature=temperature), True  # (model, is_chat)
 
-    elif model_choice == "DeepSeek":
+    elif model_choice == "Gemma-2-3B":
+        llm = HuggingFaceEndpoint(
+            repo_id="google/gemma-2-2b-it",
+            huggingfacehub_api_token=api_key,
+            task="text-generation",
+            max_new_tokens=500
+        )
+        return ChatHuggingFace(llm=llm, temperature=temperature), True
+
+    elif model_choice == "DeepSeek-685B":
         llm = HuggingFaceEndpoint(
             repo_id="deepseek-ai/DeepSeek-V3.2-Exp",
             huggingfacehub_api_token=api_key,
@@ -117,7 +126,7 @@ def build_model(model_choice, temperature=0.7):
         )
         return ChatHuggingFace(llm=llm, temperature=temperature), True
 
-    elif model_choice == "OpenAI":
+    elif model_choice == "OpenAI-20B":
         llm = HuggingFaceEndpoint(
             repo_id="openai/gpt-oss-20b",
             huggingfacehub_api_token=api_key,
@@ -148,7 +157,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", ["Flan-T5 (Free)", "DeepSeek", "OpenAI"])
+model_choice = st.radio("Model to Use", ["TinyLlama-1.1B", "Gemma-2-3B", "DeepSeek-685B", "OpenAI-20B"])
 temperature = st.slider("Temperature", 0, 100, value=50) / 100.0
 
 # Get available languages for this video