Update app.py

app.py

@@ -22,27 +22,15 @@ with st.form(key='tweet',clear_on_submit=False):
     text=st.text_area('Copy and paste a tweet or type one',placeholder='I find it quite amusing how people ignore the effects of not taking the vaccine')
     #Set examples
     alt_text=st.selectbox("Can't Type? Select an Example below",('I hate the vaccines','Vaccines made from dead human tissues','Take the vaccines or regret the consequences','Covid is a Hoax','Making the vaccines is a huge step forward for humanity. Just take them'))
-        #Submit
-    submit=st.form_submit_button('submit')
-if submit:
-    #Check text
-    if text=="":
-        text=alt_text
-        st.write(f'input text is set to {text}')    
-    else:
-        st.success('Text received',icon='✅')
-#Select a model
-exp=st.expander(label='Choose a model and Click Continue or go ahead and Click continue to use default')
-models={'Roberta': 'Junr-syl/sentiments_analysis_Roberta',
+    #Select a model
+    models={'Roberta': 'Junr-syl/sentiments_analysis_Roberta',
         'Bert':    'Junr-syl/sentiments_analysis_upgrade',
         'Distilbert':'Junr-syl/sentiments_analysis_DISTILBERT'}
-
-with exp:
     model=st.selectbox('Which model would you want to Use?',('Distilbert','Bert','Roberta'))
-
+     #Submit
+    submit=st.form_submit_button('Predict','Continue processing input')
+    
 selected_model=models[model]
-#Click continue if no model is selected
-Cont=st.button('Continue','Continue processing input')
 
 #create columns to show outputs
 col1,col2,col3=st.columns(3)
@@ -50,7 +38,14 @@ col1.write('<h2 style="font-size: 24px;"> Sentiment Emoji </h2>',unsafe_allow_ht
 col2.write('<h2 style="font-size: 24px;"> How this user feels about the vaccine </h2>',unsafe_allow_html=True)
 col3.write('<h2 style="font-size: 24px;"> Confidence of this prediction </h2>',unsafe_allow_html=True)
 
-if Cont:
+if submit:
+    #Check text
+    if text=="":
+        text=alt_text
+        st.write(f'input text is set to {text}')    
+    else:
+        st.success('Text received',icon='✅')
+        
     #import the model
     pipe=pipeline(model=selected_model)