Spaces:
Runtime error
Runtime error
File size: 1,983 Bytes
8ae0ff2 ab9256d 628051d 8ae0ff2 628051d e4f2a60 9f870a3 8ae0ff2 e4f2a60 8ae0ff2 628051d 8ae0ff2 e4f2a60 628051d e4f2a60 628051d |
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 |
import gradio as gr
from pathlib import Path
def load_models_from_file(filename):
with open(filename, 'r') as f:
return [line.strip() for line in f]
if __name__ == "__main__":
models = load_models_from_file('models.txt')
print(models)
current_model = models[0]
text_gen1 = gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link")
models2 = [gr.Interface.load(f"models/{model}", live=True, preprocess=False) for model in models]
def text_it1(inputs):
go_t1 = text_gen1(inputs)
return go_t1
def set_model(current_model):
current_model = models[current_model]
return gr.update(label=(f"{current_model}"))
def send_it1(inputs, model_choice):
proc1 = models2[model_choice]
output1 = proc1(inputs)
return output1
css = ""
with gr.Blocks(css=css) as myface:
with gr.Row():
with gr.Tab("Tools"):
with gr.Tab("View"):
with gr.Row():
gr.Pil(label="Crop")
gr.Pil(label="Crop")
with gr.Tab("Draw"):
with gr.Row():
gr.Pil(label="Crop")
gr.ImagePaint(label="Draw")
gr.Textbox(label="", lines=8)
gr.Textbox(label="", lines=8)
with gr.Tab("Color Picker"):
with gr.Row():
gr.ColorPicker(label="Color")
gr.ImagePaint(label="Draw")
with gr.Row():
gr.Textbox(lines=4, label="Magic Textbox")
gr.Button("Generate Image")
with gr.Row():
gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model)
with gr.Row():
gr.Image(label=(f"{current_model}"))
with gr.Row():
gr.Textbox(label="Prompt Idea", lines=2)
gr.Button("Use Short Prompt")
gr.Button("Extend Idea")
myface.queue(concurrency_count=200)
myface.launch(inline=True, show_api=False, max_threads=400)
|