Update app.py

app.py

@@ -30,6 +30,16 @@ def format_predictions(preds) -> str:
         out += f"{i+1}. {item['label']} (score {item['score']:.2f})\n"
     return out
 
+st.markdown(
+    """
+    <style>
+    .stApp {
+        background: linear-gradient(to bottom right, #020420, #080f6b);
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
 
 st.markdown("""
     <div style='text-align: center;'>
@@ -38,23 +48,12 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 st.markdown("""
-    <h2 style='text-align: center; color: #e80ad8; font-family: Arial;'>
+    <h2 style='text-align: center; color: lime; font-family: Arial;'>
         🚀 arXiv paper categories predictor
     </h2>
 """, unsafe_allow_html=True)
 
-
-# CSS to change the background of the entire app
-background_color_css = """
-<style>
-.stApp {
-    background-color: black;  /* #eefcfa */
-}
-</style>
-"""
-
 st.markdown("""
-    <br><br>  <!-- Adds vertical space -->
     <p style='
         color: white;
         font-size: 20px;
@@ -71,9 +70,7 @@ title = st.text_input("Title", value="")
 
 abstract = st.text_input("Abstract", value="")
 
-query = title + '\n' + abstract
-if title or abstract:
-    st.markdown("""
+st.markdown("""
         <p style='
             color: white;
             font-size: 20px;
@@ -82,5 +79,8 @@ if title or abstract:
             Most likely categories of the paper:
         </p>
     """, unsafe_allow_html=True)
+
+query = title + '\n' + abstract
+if title or abstract:
     result = format_predictions(top_pct(model(query)[0]))
     st.write(result)