Shreyas094 commited on
Commit
9739d19
·
verified ·
1 Parent(s): 8a37f10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def count_tokens(text):
14
  return len(tokenizer.encode(text))
15
  def summarize_text(text, instructions, agent_name):
16
  print(f"{agent_name}: Starting summarization")
17
- API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x22B-Instruct-v0.1"
18
  headers = {"Authorization": f"Bearer {HUGGINGFACE_TOKEN}"}
19
  payload = {
20
  "inputs": f"{instructions}\n\nText to summarize:\n{text}",
 
14
  return len(tokenizer.encode(text))
15
  def summarize_text(text, instructions, agent_name):
16
  print(f"{agent_name}: Starting summarization")
17
+ API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
18
  headers = {"Authorization": f"Bearer {HUGGINGFACE_TOKEN}"}
19
  payload = {
20
  "inputs": f"{instructions}\n\nText to summarize:\n{text}",