RamezCh commited on
Commit
53c91db
·
verified ·
1 Parent(s): f8430d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -9,13 +9,11 @@ from tools import search_tool
9
  from retriever import game_info_tool
10
  import os
11
 
12
- HUGGINGFACEHUB_API_TOKEN = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
13
-
14
- # Generate the chat interface, including the tools
15
  llm = HuggingFaceEndpoint(
16
  repo_id="google/flan-t5-base",
17
- model_kwargs={"temperature": 0.7, "max_new_tokens": 256},
18
- huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
 
19
  )
20
 
21
  chat = ChatHuggingFace(llm=llm, verbose=True)
 
9
  from retriever import game_info_tool
10
  import os
11
 
 
 
 
12
  llm = HuggingFaceEndpoint(
13
  repo_id="google/flan-t5-base",
14
+ temperature=0.7,
15
+ max_new_tokens=256,
16
+ huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN")
17
  )
18
 
19
  chat = ChatHuggingFace(llm=llm, verbose=True)