Thomas Bartlett commited on
Commit
50fe614
·
1 Parent(s): 12fe825

use a better model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,12 +63,12 @@ class BasicAgent:
63
 
64
  self.llm = ChatHuggingFace(
65
  llm=HuggingFaceEndpoint(
66
- repo_id="meta-llama/Llama-3.1-8B-Instruct",
67
  task="text-generation",
68
  max_new_tokens=512,
69
  temperature=0.0,
70
  do_sample=False,
71
- provider="hf-inference" # Automatically select the first available provider
72
  )
73
  )
74
 
 
63
 
64
  self.llm = ChatHuggingFace(
65
  llm=HuggingFaceEndpoint(
66
+ repo_id="deepseek-ai/DeepSeek-R1",
67
  task="text-generation",
68
  max_new_tokens=512,
69
  temperature=0.0,
70
  do_sample=False,
71
+ provider="auto" # Automatically select the first available provider
72
  )
73
  )
74