Update app.py

app.py

@@ -271,12 +271,10 @@ with st.container():
         os.environ["MKL_NUM_THREADS"] = str(num_threads)
 
     # Load models once and keep them in memory
-    if 'models_loaded' not in st.session_state:
-        with st.spinner("Loading models..."):
-            tokenizer_qatc, model_qatc = load_model(qatc_model_name, QATCForQuestionAnswering, device=DEVICE)
-            tokenizer_bc, model_bc = load_model(bc_model_name, ClaimModelForClassification, is_bc=True, device=DEVICE)
-            tokenizer_tc, model_tc = load_model(tc_model_name, ClaimModelForClassification, device=DEVICE)
-            st.session_state.models_loaded = True
+    tokenizer_qatc, model_qatc = load_model(qatc_model_name, QATCForQuestionAnswering, device=DEVICE)
+    tokenizer_bc, model_bc = load_model(bc_model_name, ClaimModelForClassification, is_bc=True, device=DEVICE)
+    tokenizer_tc, model_tc = load_model(tc_model_name, ClaimModelForClassification, device=DEVICE)
+    st.session_state.models_loaded = True
 
     # Store verification history
     if 'history' not in st.session_state: