Update app.py

app.py

@@ -26,6 +26,13 @@ def init_zsl_topic_classification():
     template = "This text is about {}."
     return pipe, template
 
+@st.cache(allow_output_mutation = True)
+def init_zsl_topic_classification():
+    MODEL = 'facebook/bart-large-mnli'
+    pipe = pipeline("zero-shot-classification", model=MODEL)
+    template = "This text is about {}."
+    return pipe, template
+    
 # Model initialization    
 pipeline_summarization = init_text_summarization_model()
 pipeline_zsl, template = init_zsl_topic_classification()
@@ -37,8 +44,13 @@ uploaded_file = st.file_uploader("Choose a file")
 def get_text_from_ocr_engine(uploaded_file):
     return "This is a sample text for named entity recognition and other tasks"
 
+with st.sidebar:
+    selected_menu = option_menu("Select Option", ["Upload Document", "Extract Text", "Summarize Document", "Extract Entities","Get Answers "], 
+        menu_icon="cast", default_index=1)
+    
 
-if uploaded_file is not None:
-    ocr_text  = get_text_from_ocr_engine(uploaded_file)
-    st.write(ocr_text)
- 
+if selected_menu == "Upload Document":        
+    if uploaded_file is not None:
+        ocr_text  = get_text_from_ocr_engine(uploaded_file)
+        st.write("Upload Successful")
+