Update app.py

app.py

@@ -50,16 +50,7 @@ def predict_text(text):
     return predicted_class
 
 
-def main():
-    page_selection = st.sidebar.selectbox("Выберите страницу:", ["Классификация отзывов", "Анализ токсичности"])
-    
-    if page_selection == "Классификация отзывов":
-        page_reviews_classification()
-    elif page_selection == "Анализ токсичности":
-        page_toxicity_analysis()
 
-if __name__ == "__main__":
-    main()
 
 
 def page_reviews_classification():
@@ -121,4 +112,16 @@ def page_toxicity_analysis():
         elapsed_time = time.time() - start_time
         
         for label, prob in probs.items():
-            st.write(f"Вероятность {label}: {prob:.4f}")
+            st.write(f"Вероятность {label}: {prob:.4f}")
+
+
+def main():
+    page_selection = st.sidebar.selectbox("Выберите страницу:", ["Классификация отзывов", "Анализ токсичности"])
+    
+    if page_selection == "Классификация отзывов":
+        page_reviews_classification()
+    elif page_selection == "Анализ токсичности":
+        page_toxicity_analysis()
+
+if __name__ == "__main__":
+    main()