ui fixes

backend/aragpt.py

@@ -21,15 +21,11 @@ qa_prompt_post_year = """ في سنة: """
 
 
 def write():
-    st.markdown(
-        """
-        # Arabic Language Generation:
+    _, col1, _ = st.columns(3)
 
-        Generate Arabic text using our AraGPT2 language models. You can also try AraGPT2 Zero-Shot QA capabilities.
-        Please be patient with the Mega model :)
+    with col1:
+        col1.title("Arabic Language Generation")
 
-        """
-    )
     # Sidebar
 
     # Taken from https://huggingface.co/spaces/flax-community/spanish-gpt2/blob/main/app.py
@@ -98,7 +94,7 @@ def write():
 
     st.write("#")
 
-    col = st.beta_columns(2)
+    col = st.columns(2)
 
     col[0].image("images/AraGPT2.png", width=200)
 

backend/home.py

@@ -19,7 +19,7 @@ def write():
     """
     )
     st.markdown("#")
-    col1, col2, col3 = st.beta_columns(3)
+    col1, col2, col3 = st.columns(3)
 
     col1.write("## **AraBERT**")
     col1.image("images/arabert_logo.png", width=200)

backend/processor.py

@@ -101,7 +101,7 @@ def _desegmentword(orig_word: str) -> str:
 
 
 def write():
-    _, col1, _ = st.beta_columns(3)
+    _, col1, _ = st.columns(3)
 
     with col1:
         col1.title("Arabic Text Pre-Processor")

backend/qa.py

@@ -5,7 +5,7 @@ from .services import get_qa_answers
 
 
 def write():
-    _, col1, _ = st.beta_columns(3)
+    _, col1, _ = st.columns(3)
 
     with col1:
         st.image("images/is2alni_logo.png", width=200)