Hugging Face's logo

Spaces:
pkraman06
/
PDF_Question_Answering_Bot
Sleeping

App Files Community
pkraman06 commited on
Commit
deac4cd
·
verified ·
1 Parent(s): 17d218c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ embedding_model = HuggingFaceEmbeddings(
31
  # =====================================================
32
  llm = HuggingFaceEndpoint(
33
  repo_id="NousResearch/Llama-2-7b-chat-hf",
34
- task="text-generation", # Add this explicit task definition
35
  huggingfacehub_api_token=HF_TOKEN,
36
  temperature=0.5,
37
  max_new_tokens=512
 
31
  # =====================================================
32
  llm = HuggingFaceEndpoint(
33
  repo_id="NousResearch/Llama-2-7b-chat-hf",
34
+ task="text-generation",
35
  huggingfacehub_api_token=HF_TOKEN,
36
  temperature=0.5,
37
  max_new_tokens=512