DP-CE commited on
Commit
1c064cf
·
verified ·
1 Parent(s): baca63c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -4,7 +4,7 @@ import streamlit as st
4
  from langchain_core.prompts import PromptTemplate
5
  from langchain_core.output_parsers import StrOutputParser
6
 
7
- def get_llm_hf_inference(model_id="mistralai/Mistral-7B-Instruct-v0.3", max_new_tokens=128, temperature=0.1):
8
  token = os.getenv("HF_TOKEN") # Hugging Face Spaces injects this automatically
9
 
10
  if not token:
 
4
  from langchain_core.prompts import PromptTemplate
5
  from langchain_core.output_parsers import StrOutputParser
6
 
7
+ def get_llm_hf_inference(model_id="mistralai/Mistral-7B-Instruct-v0.3", max_new_tokens=128, temperature=0.5):
8
  token = os.getenv("HF_TOKEN") # Hugging Face Spaces injects this automatically
9
 
10
  if not token: