Update sentiment-analyser.py

sentiment-analyser.py

@@ -3,22 +3,27 @@ from transformers import pipeline
 
 st.title('Sentiment Analyser App')
 st.write('Welcome to my sentiment analysis app!')
+model_options=["sentiment-analysis", "twitter-xlm-roberta-base-sentiment"]
 
 form = st.form(key='sentiment-form')
-user_input = form.text_area(label='Enter your text to analyse')
+model_type = form.selectbox(label="Select a model", options=model_options)
+user_input = form.text_area(label='Enter your text to analyse', value="Hey how are you?")
 submit = form.form_submit_button('Submit')
 
-classifier = pipeline("sentiment-analysis")    
-result = classifier(user_input)[0]    
-label = result['label']    
-score = result['score']
+def classification(user_input, type):
+    if type=="sentiment-analysis":
+        classifier = pipeline("sentiment-analysis")   
+    elif type==""
+        classifier = pipeline("sentiment-analysis", model="cardiffnlp/twitter-xlm-roberta-base-sentiment")
+    result = classifier(user_input)[0]
+    return result
+
 
 if submit:
-    classifier = pipeline("sentiment-analysis")
-    result = classifier(user_input)[0]
-    label = result['label']
-    score = result['score']
-    if label == 'POSITIVE':
+    resultf = classification(user_input, model_type)
+    label = resultf['label']
+    score = resultf['score']
+    if label == 'POSITIVE' or 'Positive':
             st.success(f'{label} sentiment (score: {score})')
     else:
         st.error(f'{label} sentiment (score: {score})')