jrkotun commited on
Commit
b754bfe
·
verified ·
1 Parent(s): 3b146b1

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -67,7 +67,7 @@ def get_ollama_explanation(query, description):
67
  # Load LLM
68
  llm = HuggingFaceEndpoint(
69
  repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
70
- task="text-generation",
71
  huggingfacehub_api_token=os.environ["LLM_TOKEN"],
72
  )
73
 
 
67
  # Load LLM
68
  llm = HuggingFaceEndpoint(
69
  repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
70
+ task="conversational",
71
  huggingfacehub_api_token=os.environ["LLM_TOKEN"],
72
  )
73