Update app.py

app.py

@@ -31,7 +31,7 @@ import pandas as pd
 api_token = os.getenv("HF_TOKEN")
 
 
-list_llm = ["mistralai/Miceli", "mistralai/Mistral-7B-Instruct-v0.3"]  
+list_llm = ["meta-llama/Meta-Llama-3-8B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"]  
 list_llm_simple = [os.path.basename(llm) for llm in list_llm]
 
 # Load PDF document and create doc splits
@@ -68,14 +68,24 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
     progress(0.1, desc="Initializing HF tokenizer...")
     
     progress(0.5, desc="Initializing HF Hub...")
+
     
-    llm = HuggingFaceEndpoint(
-        repo_id=llm_model,
-        huggingfacehub_api_token=api_token,
-        temperature=temperature,
-        max_new_tokens=max_tokens,
-        top_k=top_k,
-    )
+    if llm_model == "meta-llama/Meta-Llama-3-8B-Instruct":
+        llm = HuggingFaceEndpoint(
+            repo_id=llm_model,
+            huggingfacehub_api_token = api_token,
+            temperature = temperature,
+            max_new_tokens = max_tokens,
+            top_k = top_k,
+        )
+    else:
+        llm = HuggingFaceEndpoint(
+            huggingfacehub_api_token = api_token,
+            repo_id=llm_model, 
+            temperature = temperature,
+            max_new_tokens = max_tokens,
+            top_k = top_k,
+        )
     
     progress(0.75, desc="Defining buffer memory...")
     memory = ConversationBufferMemory(