theekshana
commited on
Commit
•
1df85e0
1
Parent(s):
dad185d
model name
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def initialize_app():
|
|
27 |
nltk.download("punkt")
|
28 |
SESSION_DEFAULTS = {
|
29 |
"model_type": "local",
|
30 |
-
"model_name": "
|
31 |
"summarizer_type": "Map Reduce",
|
32 |
"is_parameters_changed":False,
|
33 |
# "user_question":'',
|
@@ -89,7 +89,7 @@ def side_bar():
|
|
89 |
with st.form('param_form'):
|
90 |
# st.info('Info: use openai chat model for best results')
|
91 |
model_name = st.selectbox(
|
92 |
-
"
|
93 |
MODELS,
|
94 |
# options=["long-t5 v0", "long-t5 v1", "pegasus-x-large v1", "OpenAI"],
|
95 |
key="Model Name",
|
@@ -136,17 +136,18 @@ def side_bar():
|
|
136 |
alert.empty() # Clear the alert
|
137 |
st.markdown(
|
138 |
"### How to use\n"
|
139 |
-
"1. Select the
|
140 |
-
"1. If selected model asks for a api key enter a valid api key.\n" # noqa: E501
|
141 |
-
"1. Enter
|
142 |
)
|
143 |
st.markdown("---")
|
144 |
st.markdown("""
|
145 |
This app supports text in the following formats:
|
146 |
- Raw text in text box
|
147 |
-
- URL of article/news to be summarized
|
148 |
- .txt, .pdf, .docx file formats
|
149 |
-
"""
|
|
|
|
|
150 |
|
151 |
|
152 |
def load_app():
|
|
|
27 |
nltk.download("punkt")
|
28 |
SESSION_DEFAULTS = {
|
29 |
"model_type": "local",
|
30 |
+
"model_name": "Boardpac summarizer v1",
|
31 |
"summarizer_type": "Map Reduce",
|
32 |
"is_parameters_changed":False,
|
33 |
# "user_question":'',
|
|
|
89 |
with st.form('param_form'):
|
90 |
# st.info('Info: use openai chat model for best results')
|
91 |
model_name = st.selectbox(
|
92 |
+
"Summary model",
|
93 |
MODELS,
|
94 |
# options=["long-t5 v0", "long-t5 v1", "pegasus-x-large v1", "OpenAI"],
|
95 |
key="Model Name",
|
|
|
136 |
alert.empty() # Clear the alert
|
137 |
st.markdown(
|
138 |
"### How to use\n"
|
139 |
+
"1. Select the Summarization model\n" # noqa: E501
|
140 |
+
# "1. If selected model asks for a api key enter a valid api key.\n" # noqa: E501
|
141 |
+
"1. Enter the text to get the summary."
|
142 |
)
|
143 |
st.markdown("---")
|
144 |
st.markdown("""
|
145 |
This app supports text in the following formats:
|
146 |
- Raw text in text box
|
|
|
147 |
- .txt, .pdf, .docx file formats
|
148 |
+
"""
|
149 |
+
# - URL of article/news to be summarized
|
150 |
+
)
|
151 |
|
152 |
|
153 |
def load_app():
|
config.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
MODELS = {
|
2 |
# "long-t5 v0":"IronOne-AI-Labs/long-t5-tglobal-16k-annual-reports-v0",
|
3 |
-
"
|
4 |
# "pegasus-x-large v1" : "IronOne-AI-Labs/pegasus-x-large-annual-report-QLoRA-fine-tuned-v1.1", #for tokenizer
|
5 |
# "openai" : "IronOne-AI-Labs/pegasus-x-large-annual-report-QLoRA-fine-tuned-v1.1"
|
6 |
}
|
|
|
1 |
MODELS = {
|
2 |
# "long-t5 v0":"IronOne-AI-Labs/long-t5-tglobal-16k-annual-reports-v0",
|
3 |
+
"Boardpac summarizer v1":"IronOne-AI-Labs/long-t5-tglobal-16k-annual-reports-v1",
|
4 |
# "pegasus-x-large v1" : "IronOne-AI-Labs/pegasus-x-large-annual-report-QLoRA-fine-tuned-v1.1", #for tokenizer
|
5 |
# "openai" : "IronOne-AI-Labs/pegasus-x-large-annual-report-QLoRA-fine-tuned-v1.1"
|
6 |
}
|