DivEye - PR (pickling error)

app.py

@@ -3,9 +3,23 @@ import numpy as np
 import pandas as pd
 from software import Software
 
-software = Software()
+software = None
+theme = gr.Theme.from_hub("gstaff/xkcd") 
+
+def warmup():
+    global software
+    print("[DivEye] Warming up models...")
+    software = Software()
+    print("[DivEye] Models are ready.")
 
 def detect_ai_text(text):
+    global software
+    if software is None:
+        return "❗ Models not loaded yet. Please wait for a few moments.", 0.0, pd.DataFrame({
+                "Source": ["AI Generated", "Human Written"],
+                "Probability (%)": [0, 0]
+            })
+
     text = text.strip()
     if not text or len(text.split()) < 15:
         return (
@@ -36,8 +50,8 @@ def detect_ai_text(text):
     return message, round(ai_prob, 3), bar_data
 
 # Gradio app setup
-with gr.Blocks(title="DivEye", theme="gstaff/xkcd") as demo:
-
+with gr.Blocks(title="DivEye", theme=theme) as demo:
+    demo.on_start(warmup)
     gr.HTML("""
     <div style="display: flex; justify-content: space-between; align-items: center; padding: 1.5rem; background: #f0f4f8; border-radius: 12px; margin-bottom: 1rem;">
         <div style="text-align: left; max-width: 70%;">