Duplicate from CVH-vn1210/make_hair

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Hair Space
+emoji: 💻
+colorFrom: blue
+colorTo: indigo
+sdk: gradio
+sdk_version: 3.24.1
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: CVH-vn1210/make_hair
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
+import huggingface_hub as hf
+import os
+
+my_file = open("./style_name.txt", "r")
+  
+# reading the file
+data = my_file.read()
+  
+# replacing end splitting the text 
+# when newline ('\n') is seen.
+data_into_list = data.split("\n")[:-1]
+
+my_file.close()
+
+hf.login(token=os.environ['model_token'])
+#remember to login with token before loading model
+def text_to_hair(prompt, guidance_scale=8, num_inference_steps=30, styles=data_into_list, model_path ="DiningSystem/hair-model2"):
+    
+    pipe = StableDiffusionPipeline.from_pretrained(os.environ['bmd'], torch_dtype=torch.float16, use_auth_token=True)
+    pipe.unet.load_attn_procs(model_path)
+    pipe.to("cuda")
+    image = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).images[0]
+    #image.save(save_name) #comment if don't want to save image
+    return image #PIL format
+
+demo = gr.Interface(fn=text_to_hair, inputs=["text", gr.Slider(5, 20, value=8, label="Guidance_scale", info="Choose between 5 and 20 to improve image's content"), 
+                                             gr.Slider(20, 500, value=20, label="Num_infer_steps", info="Choose between 20 and 500 to improve image's resolution"),
+                                            gr.Dropdown(data_into_list, interactive=True, label="Some suggestion hairstyles", info="For your suggestion and reference!")],
+                    outputs="image")
+demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+diffusers 
+huggingface-hub
+torch
+transformers
+accelerate

style_name.txt

@@ -0,0 +1,65 @@
+aaron kwok hairstyle
+afro
+bald
+beehive hairstyle
+bob hair
+bouffant
+bowl cut
+bun
+caesar hair
+chonmage
+comb over
+cornrows
+crew cut
+crop
+croydon facelift
+curly hair
+curtained hair
+cute ponytails&pigtail
+devilock
+dreadlocks
+ducktail
+emo hair
+fauxhawk
+flattop
+fontange
+french braid
+french twist
+hi-top fade
+hime cut
+horseshoe flattop
+induction cut
+jeongeun kim hair
+jimmy lin hairstyle
+khokhol
+layered hair
+liberty spikes hair
+liu hulan hairstyle
+medium-length hair
+mohawk
+mop-top hair
+mullet
+odango hair
+pageboy
+perm
+pixie cut
+ponytail
+quiff
+rattail
+razor cut
+ringlet
+shag
+shoulder-length hair
+spiky hair
+the rachel
+tonsure hair
+undercut curly
+undercut hairstyle long hair
+undercut hairstyle men pompadour
+undercut hairstyle men with square angles
+undercut side part
+undercut slicked-back
+undercut tapered sides
+updo
+waist-length hair
+wave hair