KalooAI commited on
Commit
4a82a42
·
verified ·
1 Parent(s): e0377a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,7 +65,7 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
65
  # repo_id = "meta-llama/Llama-3.1-8B-Instruct"
66
  repo_id = "Qwen/Qwen2.5-7B-Instruct"
67
 
68
- llm_endpoint = HuggingFaceEndpoint(
69
  repo_id=repo_id,
70
  task="text-generation",
71
  max_new_tokens=512,
@@ -74,7 +74,7 @@ llm_endpoint = HuggingFaceEndpoint(
74
 
75
  # 2. ChatHuggingFace will correctly route this to the /v1/chat/completions
76
  # endpoint which the Llama-3.1-8B-Instruct model supports.
77
- llm = ChatHuggingFace(llm=llm_endpoint)
78
  # -------------------------
79
  # Document loader
80
  # -------------------------
 
65
  # repo_id = "meta-llama/Llama-3.1-8B-Instruct"
66
  repo_id = "Qwen/Qwen2.5-7B-Instruct"
67
 
68
+ llm = HuggingFaceEndpoint(
69
  repo_id=repo_id,
70
  task="text-generation",
71
  max_new_tokens=512,
 
74
 
75
  # 2. ChatHuggingFace will correctly route this to the /v1/chat/completions
76
  # endpoint which the Llama-3.1-8B-Instruct model supports.
77
+ # llm = ChatHuggingFace(llm=llm_endpoint)
78
  # -------------------------
79
  # Document loader
80
  # -------------------------