Spaces:
Runtime error
Runtime error
Ahmed Faiyaz
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -3,14 +3,13 @@ import torch
|
|
3 |
from diffusers import DiffusionPipeline
|
4 |
from transformers import AutoTokenizer,AutoModel
|
5 |
from diffusers.models import AutoencoderKL
|
6 |
-
|
7 |
pipeline = DiffusionPipeline.from_pretrained(
|
8 |
"CompVis/stable-diffusion-v1-4",
|
9 |
text_encoder = AutoModel.from_pretrained("csebuetnlp/banglabert"),
|
10 |
custom_pipeline="gr33nr1ng3r/Mukh-Oboyob"
|
11 |
)
|
12 |
pipeline.unet.load_attn_procs("gr33nr1ng3r/Mukh-Oboyob")
|
13 |
-
pipeline.to(device)
|
14 |
|
15 |
def diffusion(text,num_inference_steps,guidance_scale):
|
16 |
prompt = "মেয়েটির কালো চুল ছিল। মেয়েটির মুখে ভারী মেকাপ ছিল। মেয়েটির উঁচু গালের হাড় ছিল। মেয়েটির মুখ কিছুটা খোলা ছিল। মেয়েটির চেহারা ডিম্বাকৃতির। মেয়েটির চোখা নাক ছিল। মেয়েটির ঢেউ খেলানো চুল ছিল। মেয়েটির কানে দুল পরা ছিল। মেয়েটির লিপস্টিক পরা ছিল। "
|
|
|
3 |
from diffusers import DiffusionPipeline
|
4 |
from transformers import AutoTokenizer,AutoModel
|
5 |
from diffusers.models import AutoencoderKL
|
6 |
+
|
7 |
pipeline = DiffusionPipeline.from_pretrained(
|
8 |
"CompVis/stable-diffusion-v1-4",
|
9 |
text_encoder = AutoModel.from_pretrained("csebuetnlp/banglabert"),
|
10 |
custom_pipeline="gr33nr1ng3r/Mukh-Oboyob"
|
11 |
)
|
12 |
pipeline.unet.load_attn_procs("gr33nr1ng3r/Mukh-Oboyob")
|
|
|
13 |
|
14 |
def diffusion(text,num_inference_steps,guidance_scale):
|
15 |
prompt = "মেয়েটির কালো চুল ছিল। মেয়েটির মুখে ভারী মেকাপ ছিল। মেয়েটির উঁচু গালের হাড় ছিল। মেয়েটির মুখ কিছুটা খোলা ছিল। মেয়েটির চেহারা ডিম্বাকৃতির। মেয়েটির চোখা নাক ছিল। মেয়েটির ঢেউ খেলানো চুল ছিল। মেয়েটির কানে দুল পরা ছিল। মেয়েটির লিপস্টিক পরা ছিল। "
|