test006 / app.py
Karumoon's picture
Update app.py
9daeac6
raw
history blame
3.87 kB
import gradio as gr
import os
import sys
from pathlib import Path
import time
import random
from PIL import Image
models =[
"",
"CompVis/stable-diffusion-v1-4",
"runwayml/stable-diffusion-v1-5",
"prompthero/openjourney",
#4
"stabilityai/stable-diffusion-2-1",
"stabilityai/stable-diffusion-2-1-base",
"andite/anything-v4.0",
"Linaqruf/anything-v3.0",
"eimiss/EimisAnimeDiffusion_1.0v",
"nitrosocke/Nitro-Diffusion",
#10
"wavymulder/portraitplus",
"22h/vintedois-diffusion-v0-1",
"dreamlike-art/dreamlike-photoreal-2.0",
#11
"dreamlike-art/dreamlike-diffusion-1.0",
"wavymulder/Analog-Diffusion",
"nitrosocke/redshift-diffusion",
"claudfuen/photorealistic-fuen-v1",
"prompthero/openjourney-v2",
"johnslegers/epic-diffusion",
"nitrosocke/Arcane-Diffusion",
"darkstorm2150/Protogen_x5.8_Official_Release",
]
model_1=models[1]
model_2=models[2]
model_3=models[3]
model_4=models[4]
model_5=models[5]
model_6=models[6]
model_7=models[7]
model_8=models[8]
model_9=models[9]
model_10=models[10]
model_11=models[11]
model_12=models[12]
model_13=models[13]
model_14=models[14]
model_15=models[15]
model_16=models[16]
model_17=models[17]
model_18=models[18]
model_19=models[19]
model_20=models[20]
#gr.Interface.load live preprocess postprocess
proc1=gr.Interface.load(f"models/{model_1}",live=False,preprocess=True, postprocess=False)
proc2=gr.Interface.load(f"models/{model_2}",live=False,preprocess=True, postprocess=False)
proc3=gr.Interface.load(f"models/{model_3}",live=False,preprocess=True, postprocess=False)
#from transformers import pipeline
#pipe = pipeline("translation", model="t5-base")
def randStr():
pp=["a","b","c","d","e","f","g","h"]
str="";
str+=random.choice(pp)+random.choice(pp)+random.choice(pp)
str+=random.choice(pp)+random.choice(pp)+random.choice(pp)
str+=random.choice(pp)+random.choice(pp)+random.choice(pp)
str+=random.choice(pp)+random.choice(pp)+random.choice(pp)
return str
def bbb22(text):
return text[::-1]
###
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
english = gr.Textbox(label="English text")
btn01 = gr.Button(value="btn01")
with gr.Column():
german = gr.Textbox(label="German Text")
with gr.Row():
img1=gr.Image()
img2=gr.Image()
img3=gr.Image()
with gr.Row():
btn2 = gr.Button(value="btn2")
btn3 = gr.Button(value="btn3")
btn4 = gr.Button(value="btn4")
def iimg(text):
print("\n\nvvv1 "+text+randStr()+"\n\n")
#img1.update(proc1("girl"))
#german.update("soijfoijf")
#btn2.click(proc1,inputs="girl "+randStr(),outputs=img1)
img01=proc1("girl "+randStr())
print("\n\nvvv2 "+text+randStr()+"\n\n")
img02=proc2("girl "+randStr())
#btn3.click(proc2,inputs="girl "+randStr(),outputs=img2)
print("\n\nvvv3 "+text+randStr()+"\n\n")
img03=proc3("girl "+randStr())
#btn4.click(proc3,inputs="girl "+randStr(),outputs=img3)
print("\n\nvvv4 "+text+randStr()+"\n\n")
return {img1:img1.update(img01),img2:img2.update(img02),img3:img3.update(img03),german:"aaaaaddddd"}#pipe(text)[0]["translation_text"]
#btn2.click(bbb22,inputs=english,outputs=german)
btn01.click(iimg, inputs=english, outputs=[img1,img2,img3,german])
ppStr=("girl "+randStr()+" "+english)
btn2.click(proc1,inputs=ppStr,outputs=img1)
btn3.click(proc2,inputs=ppStr,outputs=img2)
btn4.click(proc3,inputs=ppStr,outputs=img3)
examples = gr.Examples(examples=["I went to the supermarket yesterday.", "Helen is a good swimmer."],
inputs=[english])
print("\nabc01aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa\n")
demo.launch()