Update app.py

app.py

@@ -94,7 +94,6 @@ class AspectSentimentPipeline(Pipeline):
     
         aspects = []
         current_aspect = []
-        previous_b_a_index = None
         
         for i, (token, prediction) in enumerate(zip(aligned_tokens, aligned_predictions)):
             if prediction == "B-A":
@@ -102,17 +101,19 @@ class AspectSentimentPipeline(Pipeline):
                     aspects.append(" ".join(current_aspect))
                     current_aspect = []
                 current_aspect.append(token)
-                previous_b_a_index = i
             elif prediction == "I-A":
-                if previous_b_a_index is not None:
-                    aligned_tokens[previous_b_a_index] += " " + token
+                if current_aspect:
+                    current_aspect.append(token)
                 else:
                     current_aspect.append(token)
+            elif prediction == "O" or i == len(aligned_tokens) - 1:
+                if current_aspect:
+                    aspects.append(" ".join(current_aspect))
+                    current_aspect = []
             else:
                 if current_aspect:
                     aspects.append(" ".join(current_aspect))
                     current_aspect = []
-                previous_b_a_index = None
         
         if current_aspect:
             aspects.append(" ".join(current_aspect))