Sourudra commited on
Commit
069fecb
·
verified ·
1 Parent(s): 363fcc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -1
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
- preloaded_demo = gr.load("models/ZB-Tech/Text-to-Image")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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