Update app.py

app.py

@@ -19,8 +19,8 @@ except ImportError:
     logging.info("PEFT library not found. Loading models without PEFT support.")
 
 # --- Configuration ---
-# Model path is set to sojka2
-MODEL_PATH = os.getenv("MODEL_PATH", "speakleash/sojka2") 
+# Model path is set to sojka
+MODEL_PATH = os.getenv("MODEL_PATH", "AndromedaPL/sojka") 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 LABELS = ["self-harm", "hate", "vulgar", "sex", "crime"]
 MAX_SEQ_LENGTH = 512
@@ -133,13 +133,12 @@ def gradio_predict(text: str) -> Tuple[str, Dict[str, float]]:
         verdict = "✅ Komunikat jest bezpieczny."
     else:
         highest_unsafe_category = max(unsafe_categories, key=unsafe_categories.get)
-        verdict = f"⚠️ Wykryto potencjalnie szkodliwe treści w kategorii: {highest_unsafe_category.upper()}"
+        verdict = f"⚠️ Wykryto potencjalnie szkodliwe treści\n: {highest_unsafe_category.upper()}"
     
     return verdict, predictions
 
 # --- Gradio Interface ---
 
-# Custom theme inspired by the provided image - THIS IS THE CORRECTED LINE
 theme = gr.themes.Default(
     primary_hue=gr.themes.colors.blue,
     secondary_hue=gr.themes.colors.indigo,
@@ -187,7 +186,7 @@ with gr.Blocks(theme=theme, css=".gradio-container {max-width: 960px !important;
                     Bielik Guard
                 </p>
                 <h1 style="font-size: 2.8rem; font-weight: 800; line-height: 1.2; margin-top: 1rem; margin-bottom: 1rem; color: var(--neutral-800);">
-                Naucz <span style="color: var(--primary-600);">SÓJKĘ</span> – Bielik Guard dla bezpiecznej komunikacji
+                Przetestuj <span style="color: var(--primary-600);">SÓJKĘ</span> – Bielik Guard dla bezpiecznej komunikacji
                 </h1>
                 <p style="font-size: 1rem; color: var(--neutral-600); margin-bottom: 2rem;">
                 Sójka to model AI, który wykrywa i blokuje szkodliwe treści w komunikacji cyfrowej. Chroni użytkowników jak czujny strażnik swoich domów.