aniket47 commited on
Commit
cf9768a
·
1 Parent(s): b9ff913

Fix: Remove non-existent initialize_ai_model() method call

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -311,16 +311,13 @@ def main():
311
  st.success("Documents cleared!")
312
  st.rerun()
313
 
314
- # Web search status
315
  st.subheader("AI Model Status")
316
  if st.session_state.chatbot.hf_client and st.session_state.chatbot.hf_client.is_available():
317
  st.success("✅ AI model loaded")
318
  else:
319
- st.error(" AI model not loaded")
320
- if st.button("🔄 Load AI Model", type="primary"):
321
- success = st.session_state.chatbot.initialize_ai_model()
322
- if success:
323
- st.rerun()
324
 
325
  st.subheader("Web Search")
326
  if st.session_state.chatbot.web_searcher:
 
311
  st.success("Documents cleared!")
312
  st.rerun()
313
 
314
+ # AI Model status
315
  st.subheader("AI Model Status")
316
  if st.session_state.chatbot.hf_client and st.session_state.chatbot.hf_client.is_available():
317
  st.success("✅ AI model loaded")
318
  else:
319
+ st.warning("⚠️ AI model loading...")
320
+ st.info("Models are being downloaded. This may take a few minutes on first run.")
 
 
 
321
 
322
  st.subheader("Web Search")
323
  if st.session_state.chatbot.web_searcher: