redact text

app.py

@@ -112,8 +112,14 @@ if __name__ == '__main__':
                 st.error(f'Prediction: {prediction}')
 
             st.caption(f'Source: {sample["source"]} ({label})')
-            if len(sample['content']) > 300:
-                sample['content'] = sample['content'][:300]
+            # if len(sample['content']) > 300:
+            #     sample['content'] = sample['content'][:300]
+            temp = []
+            for idx, word in enumerate(sample['content'].split(' ')):
+                if idx//5 == 0:
+                    word = '*'*len(word)
+                temp.append(word)
+            sample['content'] = ' '.join(temp)
             st.markdown(sample['content'])
     else:
         title = st.text_input('Article title', 'Test title')