Update app.py

app.py

@@ -44,13 +44,19 @@ def create_db(splits):
 # Initialize langchain LLM chain
 def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
     if llm_model == "meta-llama/Meta-Llama-3-8B-Instruct":
-        llm = HuggingFaceEndpoint(
-            repo_id=llm_model,
+        # llm = HuggingFaceEndpoint(
+        #     repo_id=llm_model,
+        #     huggingfacehub_api_token = api_token,
+        #     temperature = temperature,
+        #     max_new_tokens = max_tokens,
+        #     top_k = top_k,
+        # )
+
+        llm = HuggingFaceHub(
+            repo_id = llm_model,
             huggingfacehub_api_token = api_token,
-            temperature = temperature,
-            max_new_tokens = max_tokens,
-            top_k = top_k,
-        )
+            task="text-generation",
+            )
     else:
         llm = HuggingFaceEndpoint(
             huggingfacehub_api_token = api_token,