Rename control_app.py to app.py

control_app.py → app.py

@@ -1,10 +1,3 @@
-'''
-!git clone https://huggingface.co/spaces/radames/SPIGA-face-alignment-headpose-estimator
-!cp -r SPIGA-face-alignment-headpose-estimator/SPIGA .
-!pip install -r SPIGA/requirements.txt
-!pip install datasets
-!huggingface-cli login
-'''
 from pred_color import *
 import gradio as gr
 
@@ -73,59 +66,3 @@ examples=example_sample if example_sample else None,
 )
 gr.launch(share=False)
 
-if __name__ == "__main__":
-    '''
-    control_image = load_image("./conditioning_image_1.png")
-    prompt = "戴眼镜的中年男子"
-    # generate image
-    generator = torch.manual_seed(0)
-    image = pipe(
-         prompt, num_inference_steps=50, generator=generator, image=control_image
-    ).images[0]
-    image
-
-    control_image = load_image("./conditioning_image_1.png")
-    prompt = "穿蓝色衣服的秃头男子"
-    # generate image
-    generator = torch.manual_seed(0)
-    image = pipe(
-         prompt, num_inference_steps=50, generator=generator, image=control_image
-    ).images[0]
-    image
-
-    control_image = load_image("./conditioning_image_2.png")
-    prompt = "金色头发的美丽女子"
-    # generate image
-    generator = torch.manual_seed(0)
-    image = pipe(
-         prompt, num_inference_steps=50, generator=generator, image=control_image
-    ).images[0]
-    image
-
-    control_image = load_image("./conditioning_image_2.png")
-    prompt = "绿色运动衫的男子"
-    # generate image
-    generator = torch.manual_seed(0)
-    image = pipe(
-         prompt, num_inference_steps=50, generator=generator, image=control_image
-    ).images[0]
-    image
-
-    from huggingface_hub import HfApi
-    hf_api = HfApi()
-
-    hf_api.upload_file(
-        path_or_fileobj = "TSD_save_only/diffusion_pytorch_model.bin",
-        path_in_repo = "diffusion_pytorch_model.bin",
-        repo_id = "svjack/ControlNet-Face-Zh",
-        repo_type = "model",
-    )
-
-    hf_api.upload_file(
-        path_or_fileobj = "TSD_save_only/config.json",
-        path_in_repo = "config.json",
-        repo_id = "svjack/ControlNet-Face-Zh",
-        repo_type = "model",
-    )
-    '''
-    pass