Spaces:

bohmian
/

esg_countries_chatbot

Sleeping

bohmian commited on Jun 3

Commit

3fee323

•

1 Parent(s): e736d49

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -182,8 +182,8 @@ def get_llm():
         endpoint_url=st.session_state['model'],
         huggingfacehub_api_token=os.environ['HUGGINGFACEHUB_API_TOKEN'],
         task="text-generation",
-        "temperature": st.session_state['temperature'],
-        "max_new_tokens": st.session_state['max_new_tokens']
     )
     return llm

         endpoint_url=st.session_state['model'],
         huggingfacehub_api_token=os.environ['HUGGINGFACEHUB_API_TOKEN'],
         task="text-generation",
+        temperature = st.session_state['temperature'],
+        max_new_tokens = st.session_state['max_new_tokens']
     )
     return llm