Update app.py

app.py

@@ -31,10 +31,11 @@ def summarize_text(text):
     resp = summarizer(text)
     return resp[0]['summary_text']
 
-def text_to_sentiment(text, all_score=False):
-    sentiment = fin_model(text, return_all_scores=all_score)[0]["label"]
-    return sentiment
-
+def text_to_sentiment(text, all_score=False, label == True):
+    if label:
+        return fin_model(text, return_all_scores=all_score)[0]["label"]
+    else:
+        return fin_model(text, return_all_scores=all_score)[0]
 def fin_ext(text):
     results = fin_model(split_in_sentences(text))
     return make_spans(text, results)
@@ -80,7 +81,7 @@ def process_paragraph_1_sent(paragraph):
                 try:
                     paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
                     selected_paragraph = stored_paragraphs_1[paragraph_index]
-                    results = text_to_sentiment(selected_paragraph, True)
+                    results = text_to_sentiment(selected_paragraph, True, False)
                     if isinstance(results, list) and isinstance(results[0], list):
                         # We unpack the list of dictionaries to get all labels
                         output = {result['label']: result['score'] for result in results[0]}
@@ -117,7 +118,7 @@ def process_paragraph_2_sent(paragraph):
                 try:
                     paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
                     selected_paragraph = stored_paragraphs_2[paragraph_index]
-                    results = text_to_sentiment(selected_paragraph, True)
+                    results = text_to_sentiment(selected_paragraph, True, False)
                     if isinstance(results, list) and isinstance(results[0], list):
                         # We unpack the list of dictionaries to get all labels
                         output = {result['label']: result['score'] for result in results[0]}