theekshana commited on
Commit
1df85e0
1 Parent(s): dad185d

model name

Browse files
Files changed (2) hide show
  1. app.py +8 -7
  2. config.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def initialize_app():
27
  nltk.download("punkt")
28
  SESSION_DEFAULTS = {
29
  "model_type": "local",
30
- "model_name": "long-t5 v1",
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
- "Chat model",
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 LLM 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 a text or a url to get a summary."
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
- "long-t5 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
  }
 
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
  }