diff --git a/server.py b/server.py
index ac3bcec1..fc99ef72 100644
--- a/server.py
+++ b/server.py
@@ -208,6 +208,7 @@ if __name__ == "__main__":
         p = Path(shared.model_name)
         if p.exists():
             model_name = p.parts[-1]
+            shared.model_name = model_name
         else:
             model_name = shared.model_name