m

entklei.py

@@ -18,6 +18,8 @@ import os
 import pydash as _
 import boto3
 
+is_production = True
+
 age_detection_model = ViTForImageClassification.from_pretrained(
     'nateraw/vit-age-classifier')
 age_detection_transforms = ViTFeatureExtractor.from_pretrained(
@@ -131,8 +133,7 @@ base_pipe.safety_checker = None
 
 pipe_with_tit_slider = _.clone_deep(base_pipe)
 
-pipe_with_tit_slider.load_lora_weights(os.path.join(os.environ.get(
-        'path', "."), "models", "breastsizeslideroffset.safetensors"), weight_name="breastsizeslideroffset.safetensors", adapter_name="breastsizeslideroffset")
+pipe_with_tit_slider.load_lora_weights(os.path.join("/repository" if is_production else ".", "models", "breastsizeslideroffset.safetensors"), weight_name="breastsizeslideroffset.safetensors", adapter_name="breastsizeslideroffset")
 
 pipe_with_small_tits = _.clone_deep(pipe_with_tit_slider)
 pipe_with_small_tits.set_adapters("breastsizeslideroffset", adapter_weights=[-0.8])