Update app.py

app.py

@@ -17,8 +17,11 @@ stop_event = threading.Event()
 
 def initialize_model():
     """Initialize Whisper model and AI detectors."""
-    st.session_state.model = WhisperModel(model_size="small", device="cpu", compute_type="int8")
-    st.session_state.ai_detector = pipeline('text-classification', model='roberta-base-openai-detector')
+    try:
+        st.session_state.model = WhisperModel("small", device="cpu", compute_type="int8")
+        st.session_state.ai_detector = pipeline('text-classification', model='roberta-base-openai-detector')
+    except Exception as e:
+        st.error(f"Error initializing models: {str(e)}")
 
 def advanced_ai_detection(text, ai_detector):
     """Perform AI detection on the text."""