iShare commited on
Commit
cacdd22
·
1 Parent(s): 8e2baa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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