Update app.py

app.py

@@ -29,10 +29,19 @@ def load_model_and_tokenizer():
     with open(config_path, "r") as f:
         config = json.load(f)
 
+    # Get vocabulary size from the tokenizer
+    vocab_size = len(tokenizer.word_index) + 1  # +1 for padding token
+
+    # Initialize embedding matrix (vocab_size x embed_size)
+    embedding_matrix = torch.zeros((vocab_size, config["embedding_dim"]))
+
     # Load model
     model = BiLSTM(
-        embedding_matrix=torch.zeros((config["max_length"], config["embedding_dim"])),
-        num_classes=config["num_labels"]
+        embedding_matrix=embedding_matrix,
+        embed_size=config["embedding_dim"],
+        num_labels=config["num_labels"],
+        hidden_size=config["hidden_size"],
+        dropout=config["dropout_rate"]
     )
     model.load_state_dict(torch.load(model_weights_path, map_location=DEVICE))
     model.eval()