Spaces:
Build error
Build error
File size: 406 Bytes
7009660 578a5f5 7009660 578a5f5 7009660 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# %%
import load_model
import cloudpickle
import os
with open('.openapikey', 'r') as reader:
os.environ['OPEN_API_KEY']=reader.read()
# %%
# llm = load_model.load_gpu_model("decapoda-research/llama-7b-hf")
llm= load_model.load_openai_model()
# %%
chain = load_model.create_chain(llm, collection="axa_terms_hkunlpinstructorlarge")
result = chain({"query": "What are AXAs green Goals?"})
print(result)
|