Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -68,7 +68,23 @@ def custom_model_ui():
|
|
68 |
def preloaded_model_ui():
|
69 |
with gr.Blocks() as preloaded_demo:
|
70 |
gr.Markdown("## Quick and Efficient Text-to-Image Generation")
|
71 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
|
73 |
return preloaded_demo
|
74 |
|
|
|
68 |
def preloaded_model_ui():
|
69 |
with gr.Blocks() as preloaded_demo:
|
70 |
gr.Markdown("## Quick and Efficient Text-to-Image Generation")
|
71 |
+
|
72 |
+
# Simulate a preloaded model without showing its name
|
73 |
+
def preloaded_infer(prompt, seed):
|
74 |
+
generator = torch.Generator().manual_seed(seed)
|
75 |
+
image = pipe(prompt=prompt, generator=generator, guidance_scale=7.5, num_inference_steps=30).images[0]
|
76 |
+
return image, seed
|
77 |
+
|
78 |
+
prompt = gr.Textbox(label="Prompt")
|
79 |
+
seed = gr.Slider(0, MAX_SEED, label="Seed", step=1, value=0)
|
80 |
+
generate_button = gr.Button("Generate")
|
81 |
+
result_image = gr.Image(label="Generated Image")
|
82 |
+
|
83 |
+
generate_button.click(
|
84 |
+
preloaded_infer,
|
85 |
+
inputs=[prompt, seed],
|
86 |
+
outputs=[result_image, seed]
|
87 |
+
)
|
88 |
|
89 |
return preloaded_demo
|
90 |
|