File size: 1,833 Bytes
4976ff3
013155b
3a598fb
 
af8341a
3a598fb
 
af8341a
3a598fb
f46ba2f
e6f363e
f505125
 
3a598fb
af8341a
3a598fb
 
 
af8341a
3a598fb
af8341a
3a598fb
 
 
802f5e1
3a598fb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26


import random
import re
import torch
import requests
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
from torch import autocast
import gradio as gr

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
CUDA_VISIBLE_DEVICES=1

def txt2img(prompt):

  image = pipe(prompt +",γ€€realistic, highly detailed, high quality", height=512, width=512, negative_prompt = "((low quality)),((poor quality)),((clone)),retro style, bad anatomy,((lowres)), blurry, (worst quality), ((low quality)), normal quality,bad anatomy, disfigured, deformed, mutation, mutilated, ugly, totem pole,(poorly drawn face), cloned face, several faces, long neck, mutated hands, bad hands, poorly drawn hands,extra limbs, malformed limbs, missing arms, missing fingers, extra fingers, fused fingers, too many fingers,missing legs, extra legs, malformed legs, extra digit, fewer digits, glitchy, cropped, jpeg artifacts, signature, watermark, username, text, errorretro style ,bad anatomy,((lowres)), blurry, (worst quality), normal quality,bad anatomy, disfigured, deformed, mutation, mutilated, ugly, totem pole,(poorly drawn face), cloned face, several faces, long neck, mutated hands, bad hands, poorly drawn hands,extra limbs, malformed limbs, missing arms, missing fingers, extra fingers, fused fingers, too many fingers,missing legs, extra legs, malformed legs, extra digit, fewer digits, glitchy, cropped, jpeg artifacts, signature, watermark, username, text, error", guidance_scale = 7.5,num_inference_steps = 50).images[0]
    
  image.save("sd_image.png")

  return image

pipe = StableDiffusionPipeline.from_pretrained("Fung804/makoto-shinkai-v2", torch_dtype=torch.float16)  
pipe = pipe.to("cuda")    
generate = gr.Interface(fn = txt2img, inputs="text",outputs="image",allow_flagging="never")

generate.launch()