Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -25,12 +25,11 @@ from aspect_ratio_template import aspect_ratios
|
|
25 |
# model_id = 'SG161222/RealVisXL_V5.0'
|
26 |
# model_id = 'Lykon/dreamshaper-xl-lightning'
|
27 |
model_id = 'SG161222/RealVisXL_V5.0_Lightning'
|
28 |
-
# model_file = "https://huggingface.co/RunDiffusion/Juggernaut-XI-v11/blob/main/Juggernaut-XI-byRunDiffusion.safetensors"
|
29 |
# model_id = 'RunDiffusion/Juggernaut-XI-v11'
|
|
|
30 |
base_model_path = Path(model_id)
|
31 |
os.makedirs(base_model_path, exist_ok=True)
|
32 |
snapshot_download(repo_id=model_id, local_dir=base_model_path)
|
33 |
-
#model_file = hf_hub_download(repo_id=model_id, filename="Juggernaut-XI-byRunDiffusion.safetensors", repo_type="model")
|
34 |
|
35 |
face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
|
36 |
face_detector.prepare(ctx_id=0, det_size=(640, 640))
|
|
|
25 |
# model_id = 'SG161222/RealVisXL_V5.0'
|
26 |
# model_id = 'Lykon/dreamshaper-xl-lightning'
|
27 |
model_id = 'SG161222/RealVisXL_V5.0_Lightning'
|
|
|
28 |
# model_id = 'RunDiffusion/Juggernaut-XI-v11'
|
29 |
+
|
30 |
base_model_path = Path(model_id)
|
31 |
os.makedirs(base_model_path, exist_ok=True)
|
32 |
snapshot_download(repo_id=model_id, local_dir=base_model_path)
|
|
|
33 |
|
34 |
face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
|
35 |
face_detector.prepare(ctx_id=0, det_size=(640, 640))
|