Update imagefunc.py

This commit is contained in:
Xumin Yu
2024-11-06 08:06:02 +08:00
committed by GitHub
parent 458456e464
commit b0b68c8194
+2 -2
View File
@@ -1626,7 +1626,7 @@ def get_a_person_mask_generator_model_path() -> str:
import wget
model_url = f'https://storage.googleapis.com/mediapipe-models/image_segmenter/selfie_multiclass_256x256/float32/latest/{model_name}'
log(f"Downloading '{model_name}' model")
os.makedirs(model_file_path, exist_ok=True)
os.makedirs(os.path.dirname(model_file_path), exist_ok=True)
wget.download(model_url, model_file_path)
return model_file_path
@@ -2503,4 +2503,4 @@ minicpm_llama3_v25_prompts = """
# OUTPUT
Your output should just be an plain list of descriptions. No numbers, no extraneous labels, no hyphens.
Create only one prompt.
"""
"""