diff --git a/1/model.py b/1/model.py
index a0b8378..57aa088 100644
--- a/1/model.py
+++ b/1/model.py
@@ -19,8 +19,8 @@ class TritonPythonModel:
 
         self.model_name = args["model_name"]
         model_repository = args["model_repository"]
-        #model_path = f"{model_repository}/{self.model_name}"
-        model_path = "/cheetah/input/model/gemma-3-1b-it/gemma-3-1b-it"
+        model_path = f"{model_repository}/{self.model_name}"
+        #model_path = "/cheetah/input/model/gemma-3-1b-it/gemma-3-1b-it"
 
         input_model_path = model_path