Update app.py

app.py

@@ -8,9 +8,9 @@ from semviqa.tvc.tvc_eval import classify_claim
 
 # Load models with caching
 @st.cache_resource()
-def load_model(model_name, model_class):
+def load_model(model_name, model_class, is_bc=False):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = model_class.from_pretrained(model_name)
+    model = model_class.from_pretrained(model_name, num_labels=3 if not is_bc else 2)
     return tokenizer, model
 
 # UI Configuration
@@ -26,7 +26,7 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-st.markdown("<p class='big-title'>🔍 SemViQA: Vietnamese Fact-Checking System</p>", unsafe_allow_html=True)
+st.markdown("<p class='big-title'>🔍 SemViQA: A Semantic Question Answering System for Vietnamese Information Fact-Checking</p>", unsafe_allow_html=True)
 st.markdown("<p class='sub-title'>Enter a claim and context to verify its accuracy</p>", unsafe_allow_html=True)
 
 # Sidebar - Configuration Settings
@@ -39,7 +39,7 @@ with st.sidebar.expander("⚙️ Settings", expanded=False):
 
 # Load selected models
 tokenizer_qatc, model_qatc = load_model(qatc_model_name, QATCForQuestionAnswering)
-tokenizer_bc, model_bc = load_model(bc_model_name, ClaimModelForClassification)
+tokenizer_bc, model_bc = load_model(bc_model_name, ClaimModelForClassification, is_bc=True)
 tokenizer_tc, model_tc = load_model(tc_model_name, ClaimModelForClassification)
 
 # User Input Fields