test006 / app.py
Karumoon's picture
Update app.py
8613774
raw
history blame
2.01 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]
proc1=gr.Interface.load(f"models/{model_1}",live=False,preprocess=True, postprocess=False)
#from transformers import pipeline
#pipe = pipeline("translation", model="t5-base")
def translate(text):
print("vvv")
#img1.update(proc1("girl"))
#german.update("soijfoijf")
img0=proc1("girl")
return img0#pipe(text)[0]["translation_text"]
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
english = gr.Textbox(label="English text")
translate_btn = gr.Button(value="btn1")
translate_btn2 = gr.Button(value="btn2")
with gr.Column():
german = gr.Textbox(label="German Text")
with gr.Row():
img1=gr.Image()
img2=gr.Image()
img3=gr.Image()
translate_btn.click(translate, inputs=english, outputs=img1, api_name="translate-to-german")
translate_btn2.click("cccc",outputs=german)
examples = gr.Examples(examples=["I went to the supermarket yesterday.", "Helen is a good swimmer."],
inputs=[english])
print("abc01")
demo.launch()