dvir-bria commited on
Commit
c4630d8
1 Parent(s): 194e790

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +87 -2
README.md CHANGED
@@ -1,5 +1,90 @@
1
  ---
2
  license: other
3
- license_name: asdasd
4
- license_link: LICENSE
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: other
3
+ license_name: bria-2.2
4
+ license_link: https://bria.ai/customer-general-terms-and-conditions
5
+ inference: false
6
+
7
+ tags:
8
+ - text-to-image
9
+ - controlnet model
10
+ - legal liability
11
+ - commercial use
12
+ extra_gated_prompt: This model weights by BRIA AI can be obtained after a commercial license is agreed upon. Fill in the form below and we reach out to you.
13
+ extra_gated_fields:
14
+ Name: text
15
+ Company/Org name: text
16
+ Org Type (Early/Growth Startup, Enterprise, Academy): text
17
+ Role: text
18
+ Country: text
19
+ Email: text
20
+ By submitting this form, I agree to BRIA’s Privacy policy and Terms & conditions, see links below: checkbox
21
  ---
22
+
23
+ # BRIA 2.2 ControlNet Canny Model Card
24
+
25
+
26
+ [***Click here for Demo***](https://huggingface.co/spaces/briaai/BRIA-2.2-ControlNets)
27
+
28
+
29
+ BRIA 2.2 ControlNet-Canny, trained on the foundation of [BRIA 2.2 Text-to-Image](https://huggingface.co/briaai/BRIA-2.2), enables the generation of high-quality images guided by a textual prompt and the extracted edge map from an input image. This allows for the creation of different variations of an image, all sharing the same geometry.
30
+
31
+
32
+ [BRIA 2.2](https://huggingface.co/briaai/BRIA-2.2) was trained from scratch exclusively on licensed data from our esteemed data partners. Therefore, they are safe for commercial use and provide full legal liability coverage for copyright and privacy infringement, as well as harmful content mitigation. That is, our dataset does not contain copyrighted materials, such as fictional characters, logos, trademarks, public figures, harmful content, or privacy-infringing content.
33
+
34
+ ![photo-4426232_collage.png](https://cdn-uploads.huggingface.co/production/uploads/6571c468b622b6c62c1ac4da/VzUtWzN0KdT7B-xoBNEcB.png)
35
+
36
+
37
+ ### Model Description
38
+
39
+ - **Developed by:** BRIA AI
40
+ - **Model type:** [ControlNet](https://huggingface.co/docs/diffusers/using-diffusers/controlnet) for Latent diffusion
41
+ - **License:** [bria-2.2](https://bria.ai/bria-huggingface-model-license-agreement/)
42
+
43
+ - **Model Description:** ControlNet Canny for BRIA 2.2 Text-to-Image model. The model generates images guided by text and the edge map of the conditioned image.
44
+ - **Resources for more information:** [BRIA AI](https://bria.ai/)
45
+
46
+
47
+ ### Get Access
48
+ BRIA 2.2 ControlNet-Canny requires access to BRIA 2.2 Text-to-Image. For more information, [click here](https://huggingface.co/briaai/BRIA-2.2).
49
+
50
+
51
+
52
+
53
+
54
+
55
+ ### Code example using Diffusers
56
+
57
+
58
+ ```
59
+ pip install diffusers
60
+ ```
61
+
62
+
63
+ ```py
64
+ from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline
65
+ import torch
66
+
67
+ controlnet = ControlNetModel.from_pretrained(
68
+ "briaai/ControlNet-Canny",
69
+ torch_dtype=torch.float16
70
+ )
71
+
72
+ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
73
+ "briaai/BRIA-2.2",
74
+ controlnet=controlnet,
75
+ torch_dtype=torch.float16,
76
+ )
77
+ pipe.to("cuda")
78
+
79
+ prompt = "A portrait of a Beautiful and playful ethereal singer, golden designs, highly detailed, blurry background"
80
+ negative_prompt = "Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
81
+
82
+ # Calculate Canny image
83
+ input_image = cv2.imread('pics/singer.png')
84
+ input_image = cv2.Canny(input_image, low_threshold, high_threshold)
85
+ input_image = input_image[:, :, None]
86
+ input_image = np.concatenate([input_image, input_image, input_image], axis=2)
87
+ canny_image = Image.fromarray(image)
88
+
89
+ image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=canny_image, controlnet_conditioning_scale=1.0, height=1024, width=1024).images[0]
90
+ ```