lopera47 commited on
Commit
848648c
·
verified ·
1 Parent(s): e9d3f86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -46,7 +46,8 @@ class BasicAgent: # Some times Inheritance is needed
46
  llm = ChatHuggingFace(
47
  repo_id="tiiuae/falcon-7b-instruct",
48
  huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
49
- llm={"temperature": 0.5, "max_new_tokens": 512}
 
50
  )
51
 
52
  # # Define the LLM
 
46
  llm = ChatHuggingFace(
47
  repo_id="tiiuae/falcon-7b-instruct",
48
  huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
49
+ temperature=0.5,
50
+ max_new_tokens=512
51
  )
52
 
53
  # # Define the LLM