fix demo

app.py

@@ -17,6 +17,7 @@ from insightface.app import FaceAnalysis
 from PIL import Image
 import numpy as np
 import random
+import os
 
 import gradio as gr
 
@@ -32,7 +33,8 @@ else:
 
 # download models
 from huggingface_hub import hf_hub_download
-from modelscope import snapshot_download
+#from modelscope import snapshot_download
+from modelscope.hub.file_download import model_file_download
 
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arc2face/config.json", local_dir="./models")
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arc2face/diffusion_pytorch_model.safetensors", local_dir="./models")
@@ -40,7 +42,16 @@ hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="encoder/config.json", lo
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="encoder/pytorch_model.bin", local_dir="./models")
 hf_hub_download(repo_id="FoivosPar/Arc2Face", filename="arcface.onnx", local_dir="./models/antelopev2")
 
-base_model = snapshot_download('AI-ModelScope/stable-diffusion-v1-5', cache_dir='./models')
+#base_model = snapshot_download('AI-ModelScope/stable-diffusion-v1-5', cache_dir='./models')
+model_dir = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='model_index.json', cache_dir='./models')
+base_model = os.path.dirname(model_dir)
+_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='scheduler/scheduler_config.json', cache_dir='./models')
+_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/merges.txt', cache_dir='./models')
+_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/special_tokens_map.json', cache_dir='./models')
+_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/tokenizer_config.json', cache_dir='./models')
+_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='tokenizer/vocab.json', cache_dir='./models')
+_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='vae/config.json', cache_dir='./models')
+_ = model_file_download(model_id='AI-ModelScope/stable-diffusion-v1-5', file_path='vae/diffusion_pytorch_model.safetensors', cache_dir='./models')
 
 # Load face detection and recognition package
 app = FaceAnalysis(name='antelopev2', root='./', providers=['CPUExecutionProvider'])
@@ -59,7 +70,8 @@ pipeline = StableDiffusionPipeline.from_pretrained(
         text_encoder=encoder,
         unet=unet,
         torch_dtype=dtype,
-        safety_checker=None
+        safety_checker=None,
+        feature_extractor=None
     )
 pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
 pipeline = pipeline.to(device)