Update app.py

app.py

@@ -82,8 +82,16 @@ def create_vector_store(transcript):
     return FAISS.from_documents(docs, load_embeddings())
 
 # 🤖 Model Builder
+# 🤖 Model Builder (with free model option)
 def build_model(model_choice, temperature):
-    repo_id = "deepseek-ai/DeepSeek-V3.2-Exp" if model_choice == "DeepSeek" else "openai/gpt-oss-20b"
+    if model_choice == "DeepSeek":
+        repo_id = "deepseek-ai/DeepSeek-V3.2-Exp"  # paid
+    elif model_choice == "OpenAI":
+        repo_id = "openai/gpt-oss-20b"             # paid
+    else:
+        # Free Hugging Face model
+        repo_id = "bigscience/bloom-560m"         # free, smaller model
+
     llm = HuggingFaceEndpoint(
         repo_id=repo_id,
         huggingfacehub_api_token=api_key,
@@ -91,6 +99,7 @@ def build_model(model_choice, temperature):
     )
     return ChatHuggingFace(llm=llm, temperature=temperature)
 
+
 # 🧾 Prompt Template
 prompt_template = PromptTemplate(
     template=(
@@ -123,7 +132,7 @@ selected_lang = st.selectbox("Transcript Language", lang_options)
 language_code = selected_lang.split("(")[-1].strip(")")
 
 query = st.text_area("Your Query", value="What is RAG?")
-model_choice = st.radio("Model to Use", ["DeepSeek", "OpenAI"])
+model_choice = st.radio("Model to Use", ["DeepSeek", "OpenAI", "Free HF Model"])
 temperature = st.slider("Temperature", 0, 100, value=50)
 
 if st.button("🚀 Run Chatbot"):