File size: 307 Bytes
1e9ebc3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from langchain_huggingface import HuggingFaceEndpoint

import env_config

hf_api_token = env_config.HF_API_TOKEN


llm = HuggingFaceEndpoint(
    repo_id='tiiuae/falcon-7b-instruct',
    huggingfacehub_api_token=hf_api_token
)

question = 'Can you still have fun'
output = llm.invoke(question)
print(output)