Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import streamlit as st
|
2 |
-
from langchain.chat_models import ChatOpenAI
|
3 |
from langchain.document_loaders import WebBaseLoader
|
4 |
from langchain.chains.summarize import load_summarize_chain
|
5 |
#from bs4 import BeautifulSoup
|
@@ -24,6 +24,7 @@ llm = HuggingFaceHub(repo_id=repo_id, # for StarChat
|
|
24 |
"temperature": 0.01,
|
25 |
"top_k": 50,
|
26 |
"top_p": 0.95, "eos_token_id": 49155})
|
|
|
27 |
chain = load_summarize_chain(llm, chain_type="stuff")
|
28 |
#chain = load_summarize_chain(llm, chain_type="refine")
|
29 |
|
@@ -53,7 +54,7 @@ if url !="" and not url.strip().isspace() and not url == "" and not url.strip()
|
|
53 |
docs = loader.load()
|
54 |
result=chain.run(docs) #这个result的格式比较特殊,可以直接print,但不可以和其他字符串联合print输出
|
55 |
print(url)
|
56 |
-
print("AI Summarization: "+result) #这个会出错,原因见上方
|
57 |
print("AI Summarization:")
|
58 |
print(result)
|
59 |
|
|
|
1 |
import streamlit as st
|
2 |
+
#from langchain.chat_models import ChatOpenAI
|
3 |
from langchain.document_loaders import WebBaseLoader
|
4 |
from langchain.chains.summarize import load_summarize_chain
|
5 |
#from bs4 import BeautifulSoup
|
|
|
24 |
"temperature": 0.01,
|
25 |
"top_k": 50,
|
26 |
"top_p": 0.95, "eos_token_id": 49155})
|
27 |
+
|
28 |
chain = load_summarize_chain(llm, chain_type="stuff")
|
29 |
#chain = load_summarize_chain(llm, chain_type="refine")
|
30 |
|
|
|
54 |
docs = loader.load()
|
55 |
result=chain.run(docs) #这个result的格式比较特殊,可以直接print,但不可以和其他字符串联合print输出
|
56 |
print(url)
|
57 |
+
#print("AI Summarization: "+result) #这个会出错,原因见上方
|
58 |
print("AI Summarization:")
|
59 |
print(result)
|
60 |
|