Merlintxu commited on
Commit
2b88805
1 Parent(s): 76a0700

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from langchain.vectorstores import FAISS
5
  from langchain.embeddings import HuggingFaceEmbeddings
6
 
7
  # Initialize the LLM from Hugging Face Hub
8
- llm = HuggingFaceHub(repo_id="gpt2")
9
 
10
  # Initialize embeddings
11
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
5
  from langchain.embeddings import HuggingFaceEmbeddings
6
 
7
  # Initialize the LLM from Hugging Face Hub
8
+ llm = HuggingFaceHub(repo_id="gpt2", model_kwargs={"temperature": 0.7})
9
 
10
  # Initialize embeddings
11
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")