From 1bf3cb79713f39ff4c2722e52059806b7a469720 Mon Sep 17 00:00:00 2001
From: Sandra Friebolin <friebolin@cluster.cl.uni-heidelberg.de>
Date: Fri, 24 Feb 2023 13:06:38 +0100
Subject: [PATCH] Update inference

---
 Code/inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Code/inference.py b/Code/inference.py
index 7689fdd..4e854d4 100644
--- a/Code/inference.py
+++ b/Code/inference.py
@@ -67,7 +67,7 @@ input_as_dataset=preprocess.tokenizer_new(tokenizer, data_sample, max_length=512
 
 # Load model
 
-model=models.WordClassificationModel.from_pretrained("bert-base-uncased")
+model=models.WordClassificationModel("bert-base-uncased")
 
 model_path = "saved_models/bert.pth"
 model = torch.load(model_path, map_location=device)
-- 
GitLab