Update app.py

app.py

@@ -94,24 +94,25 @@ class AspectSentimentPipeline(Pipeline):
     
         aspects = []
         current_aspect = []
-            
-        special_characters = ["+", "'"]
+        previous_b_a_index = None
         
-        for token, prediction in zip(aligned_tokens, aligned_predictions):
+        for i, (token, prediction) in enumerate(zip(aligned_tokens, aligned_predictions)):
             if prediction == "B-A":
                 if current_aspect:
                     aspects.append(" ".join(current_aspect))
                     current_aspect = []
                 current_aspect.append(token)
+                previous_b_a_index = i
             elif prediction == "I-A":
-                if current_aspect:
+                if previous_b_a_index is not None:
+                    aligned_tokens[previous_b_a_index] += " " + token
+                else:
                     current_aspect.append(token)
-            elif token in special_characters and current_aspect:
-                current_aspect[-1] += token
             else:
                 if current_aspect:
                     aspects.append(" ".join(current_aspect))
                     current_aspect = []
+                previous_b_a_index = None
         
         if current_aspect:
             aspects.append(" ".join(current_aspect))