diff --git a/src/aiunn/inference/inference.py b/src/aiunn/inference/inference.py
index 1ff5994..953cc0b 100644
--- a/src/aiunn/inference/inference.py
+++ b/src/aiunn/inference/inference.py
@@ -30,7 +30,7 @@ class aiuNNInference:
             self.device = device
             
         # Load the model with specified precision
-        self.model = aiuNN.from_pretrained(model_path)
+        self.model = aiuNN.from_pretrained(model_path,)
         self.model.to(self.device)
         self.model.eval()