Spaces:
Runtime error
Runtime error
File size: 1,710 Bytes
99f1917 |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import gradio as gr
import os
import torch
from image_generator import generate_image_per_prompt_style
torch.manual_seed(11)
# Set device
torch_device = (
"cuda"
if torch.cuda.is_available()
else "mps"
if torch.backends.mps.is_available()
else "cpu"
)
if "mps" == torch_device:
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
# Define Interface
title = "Generative Art - Stable Diffusion with Styles and additional guidance"
gr_interface = gr.Interface(
generate_image_per_prompt_style,
inputs=[
gr.Textbox("cat running", label="Prompt"),
gr.Dropdown(
[
"illustration_style",
"line-art",
"hitokomoru-style",
"midjourney-style",
"hanfu-anime-style",
"birb-style",
"style-of-marc-allante",
],
value="birb-style",
label="Pre-trained Styles",
),
gr.Dropdown(
[
"blue_loss",
"cosine_loss",
],
value="cosine_loss",
label="Additional guidance for image generation",
),
gr.Textbox("on a city road", label="Additional Prompt"),
],
outputs=[
gr.Gallery(
label="Generated images",
show_label=False,
elem_id="gallery",
columns=[2],
rows=[2],
object_fit="contain",
height="auto",
)
],
title=title,
examples=[
["A flying bird", "illustration_style", "blue_loss", ""],
["cat running", "on a city road", "cosine_loss", ""]
]
)
gr_interface.launch(debug=True)
|