Update app.py

app.py

@@ -98,9 +98,11 @@ class AspectSentimentPipeline(Pipeline):
         for i, (token, prediction) in enumerate(zip(aligned_tokens, aligned_predictions)):
             if prediction == "B-A":
                 if current_aspect:
+                    current_aspect.append(token)
                     aspects.append(" ".join(current_aspect))
                     current_aspect = []
-                current_aspect.append(token)
+                else:
+                    current_aspect.append(token)
             elif prediction == "I-A":
                 if current_aspect:
                     current_aspect.append(token)
@@ -110,13 +112,6 @@ class AspectSentimentPipeline(Pipeline):
                 if current_aspect:
                     aspects.append(" ".join(current_aspect))
                     current_aspect = []
-            else:
-                if current_aspect:
-                    aspects.append(" ".join(current_aspect))
-                    current_aspect = []
-        
-        if current_aspect:
-            aspects.append(" ".join(current_aspect))
     
         return aspects