Manish-4007 commited on
Commit
6bfa6c6
·
1 Parent(s): 0cd5f06

update app

Browse files
Files changed (1) hide show
  1. app.py +15 -8
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import streamlit as st
2
 
 
3
  def load_topic_transfomers():
4
  from transformers import pipeline
5
  try:
@@ -22,17 +23,23 @@ def suggest_topic(text):
22
  return result['labels']
23
 
24
  st.title("Topic Suggestion")
25
- with st.spinner("Loading Model....."):
26
- topic_classifier = load_topic_transfomers()
27
- st.success("Model_loaded")
28
 
29
- whole_text = st.text_input("Enter the text Here: ")
30
- predicted_topic = suggest_topic(whole_text)
 
 
 
31
 
32
- st.write('Suggested Topics')
33
- for i in predicted_topic:
34
- st.write(i)
35
 
 
 
 
 
 
 
 
 
36
 
37
 
38
 
 
1
  import streamlit as st
2
 
3
+ @st.cache_resource
4
  def load_topic_transfomers():
5
  from transformers import pipeline
6
  try:
 
23
  return result['labels']
24
 
25
  st.title("Topic Suggestion")
 
 
 
26
 
27
+ if 'topic_model' not in st.session_state:
28
+ with st.spinner("Loading Model....."):
29
+ topic_classifier = load_topic_transfomers()
30
+ st.success("Model_loaded")
31
+ st.session_state.topic_model = True
32
 
33
+ whole_text = st.text_input("Enter the text Here: ")
 
 
34
 
35
+ try:
36
+ predicted_topic = suggest_topic(whole_text)
37
+
38
+ st.write('Suggested Topics')
39
+ for i in predicted_topic:
40
+ st.write(i)
41
+ except Exception as e:
42
+ print("Error", e)
43
 
44
 
45