Update app.py

app.py

@@ -34,18 +34,18 @@ def get_sentiment_from_llm(review_text):
         segments=segments+split_conj(sen)
     ls_outputs=[]
 
+    segments= [x if len(x)>=5 for x in segments]
     print(segments)
     for seg in segments:
-        if len(seg)>=5:
-            input = make_prompt(seg)
-            inputs = tokenizer(input, return_tensors='pt')
-            output = tokenizer.decode(
-            model.generate(
-                inputs["input_ids"],
-                max_new_tokens=100,
-            )[0],
-              skip_special_tokens=True)
-            ls_outputs.append("\n".join(output.split('PPrint ')))
+        input = make_prompt(seg)
+        inputs = tokenizer(input, return_tensors='pt')
+        output = tokenizer.decode(
+        model.generate(
+            inputs["input_ids"],
+            max_new_tokens=100,
+        )[0],
+          skip_special_tokens=True)
+        ls_outputs.append("\n".join(output.split('PPrint ')))
     return "\n".join(ls_outputs)
 demo = gr.Blocks()
 sentiment_extr = gr.Interface(