Update app.py

app.py

@@ -11,8 +11,6 @@ from datasets import load_dataset
 from sentence_transformers import SentenceTransformer
 from Scraper import Scrap
 
-st.set_page_config(layout="wide")
-
 model_checkpoint = "Rifky/indobert-hoax-classification"
 base_model_checkpoint = "indobenchmark/indobert-base-p1"
 data_checkpoint = "Rifky/indonesian-hoax-news"
@@ -70,20 +68,19 @@ if submit:
             ).flatten()
             sorted = np.argsort(similarity_score)[::-1].tolist()
             
-            input_column, reference_column = st.columns(2)
-            
-            input_column.markdown(f"<small>Compute Finished in {int(time.time() - last_time)} seconds</small>", unsafe_allow_html=True)
+            st.markdown(f"<small>Compute Finished in {int(time.time() - last_time)} seconds</small>", unsafe_allow_html=True)
             prediction = np.argmax(result, axis=-1)
             if prediction == 0:
-                input_column.success(f"This news is {label[prediction]}.")
+                st.success(f"This news is {label[prediction]}.")
             else:
-                input_column.error(f"This news is {label[prediction]}.")
+                st.error(f"This news is {label[prediction]}.")
                 
-            input_column.text(f"{int(result[prediction]*100)}% confidence")
-            input_column.progress(result[prediction])
+            st.text(f"{int(result[prediction]*100)}% confidence")
+            st.progress(result[prediction])
             
-            for i in sorted[:5]:
-                reference_column.write(f"""
-                <small>{data["url"][i].split("/")[2]}</small>
-                <a href={data["url"][i]}><h5>{data["title"][i]}</h5></a>
-                """, unsafe_allow_html=True)
+            with st.expander("Related Articles"):
+                for i in sorted[:5]:
+                    st.write(f"""
+                    <small>{data["url"][i].split("/")[2]}</small>
+                    <a href={data["url"][i]}>{data["title"][i]}</a>
+                    """, unsafe_allow_html=True)