mery22 commited on
Commit
9539980
1 Parent(s): 8a0aa99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,13 +24,13 @@ from datasets import load_dataset
24
 
25
  import transformers
26
  repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
27
-
28
  from huggingface_hub import login
29
  login(token=st.secrets["HF_TOKEN"])
30
  repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
31
 
32
- llm = HuggingFaceEndpoint(
33
- repo_id=repo_id, max_length=128, temperature=0.5, token=st.secrets["HF_TOKEN"]
34
  )
35
 
36
  from langchain_community.document_loaders import TextLoader
 
24
 
25
  import transformers
26
  repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
27
+ token=st.secrets["HF_TOKEN"]
28
  from huggingface_hub import login
29
  login(token=st.secrets["HF_TOKEN"])
30
  repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
31
 
32
+ model = HuggingFaceEndpoint(
33
+ repo_id=repo_id, max_length=128, temperature=0.5, token=token
34
  )
35
 
36
  from langchain_community.document_loaders import TextLoader