update app

app.py

@@ -2,6 +2,7 @@ import pandas as pd
 import streamlit as st
 from streamlit_text_rating.st_text_rater import st_text_rater
 from sentiment import classify_sentiment
+from zeroshot_clf import zero_shot_classification
 
 st.set_page_config(  # Alternate names: setup_page, page, layout
     layout="wide",  # Can be "centered" or "wide". In the future also "dashboard", etc.
@@ -9,6 +10,7 @@ st.set_page_config(  # Alternate names: setup_page, page, layout
     page_title='None',  # String or None. Strings get appended with "• Streamlit".
 )
 
+
 padding_top = 0
 st.markdown(f"""
     <style>
@@ -77,3 +79,12 @@ if select_task=='Detect Sentiment':
             else:
                 response = st_text_rater(t + f"--> This statement is {sentiments[i]}",
                                          color_background='rgb(233, 116, 81)',key=t)
+
+if select_task=='Zero Shot Classification':
+    st.header("You are now performing Zero Shot Classification")
+    input_texts = st.text_input(label="Input text to classify into topics")
+    input_lables = st.text_input(label="Enter labels separated by commas")
+
+    if len(input_texts) !=0 and len(input_lables) != 0:
+        output=zero_shot_classification(input_texts, input_lables)
+        st.write(output)