|
from langchain.chat_models import ChatOpenAI |
|
from langchain.document_loaders import WebBaseLoader |
|
from langchain.chains.summarize import load_summarize_chain |
|
from bs4 import BeautifulSoup |
|
import os |
|
from dotenv import load_dotenv |
|
load_dotenv() |
|
from langchain import HuggingFaceHub |
|
import requests |
|
import sys |
|
|
|
|
|
|
|
hf_token = os.environ.get('HUGGINGFACEHUB_API_TOKEN') |
|
|
|
repo_id=os.environ.get('repo_id') |
|
|
|
|
|
llm = HuggingFaceHub(repo_id=repo_id, |
|
|
|
huggingfacehub_api_token=hf_token, |
|
model_kwargs={ |
|
"min_length":1024, |
|
"max_new_tokens":3072, "do_sample":True, |
|
|
|
"temperature":0.1, |
|
"top_k":50, |
|
"top_p":0.95, "eos_token_id":49155}) |
|
|
|
loader = WebBaseLoader("https://www.usinoip.com/") |
|
docs = loader.load() |
|
|
|
|
|
chain = load_summarize_chain(llm, chain_type="stuff") |
|
|
|
result=chain.run(docs) |
|
print(result) |