Debugging 4

app.py

@@ -51,16 +51,33 @@ st.set_page_config(
 import pandas as pd
 from final import *
 
-# Cache model loading
+# # Cache model loading
+# @st.cache_resource
+# def initialize_models():
+#     nlp, tokenizer, model = load_models()
+#     knowledge_graph = load_knowledge_graph()
+#     return nlp, tokenizer, model, knowledge_graph
+
+# # Initialize all models
+# nlp, tokenizer, model, knowledge_graph = initialize_models()
+
+# Update the initialize_models function
 @st.cache_resource
 def initialize_models():
-    nlp, tokenizer, model = load_models()
+    try:
+        nlp = spacy.load("en_core_web_sm")
+    except:
+        spacy.cli.download("en_core_web_sm")
+        nlp = spacy.load("en_core_web_sm")
+    
+    model_path = "./results/checkpoint-753"
+    tokenizer = DebertaV2Tokenizer.from_pretrained('microsoft/deberta-v3-small')
+    model = AutoModelForSequenceClassification.from_pretrained(model_path)
+    model.eval()
+    
     knowledge_graph = load_knowledge_graph()
     return nlp, tokenizer, model, knowledge_graph
 
-# Initialize all models
-nlp, tokenizer, model, knowledge_graph = initialize_models()
-
 # Streamlit UI
 def main():
     st.title("📰 Nexus NLP News Classifier")

final.py

@@ -293,7 +293,7 @@ def load_models():
 
 def load_knowledge_graph():
     """Load and initialize knowledge graph"""
-    graph_path = "./models/knowledge_graph.pkl"
+    graph_path = "./knowledge_graph_final.pkl"
     with open(graph_path, 'rb') as f:
         graph_data = pickle.load(f)
     knowledge_graph = nx.DiGraph()