diff --git a/tdc/model_server/tdc_hf.py b/tdc/model_server/tdc_hf.py
index 5cf76051..223012d1 100644
--- a/tdc/model_server/tdc_hf.py
+++ b/tdc/model_server/tdc_hf.py
@@ -57,7 +57,8 @@ def load(self):
             raise Exception("this model is not in the TDC model hub GH repo.")
         elif self.model_name == "Geneformer":
             from transformers import AutoModelForMaskedLM
-            model = AutoModelForMaskedLM.from_pretrained("ctheodoris/Geneformer")
+            model = AutoModelForMaskedLM.from_pretrained(
+                "ctheodoris/Geneformer")
             return model
         elif self.model_name == "scGPT":
             from transformers import AutoModel