Update app.py

app.py

@@ -41,7 +41,7 @@ def custom_infer(
 # Gradio interface for custom model
 def custom_model_ui():
     with gr.Blocks() as custom_demo:
-        gr.Markdown("## Advanced Text-to-Image Generation")
+        gr.Markdown("## Needs a GPU for best performance and it is highly customizable.")
         with gr.Row():
             prompt = gr.Text(label="Prompt")
             run_button = gr.Button("Generate")
@@ -67,34 +67,18 @@ def custom_model_ui():
 # Preloaded Gradio model
 def preloaded_model_ui():
     with gr.Blocks() as preloaded_demo:
-        gr.Markdown("## Quick and Efficient Text-to-Image Generation")
-        
-        # Simulate a preloaded model without showing its name
-        def preloaded_infer(prompt, seed):
-            generator = torch.Generator().manual_seed(seed)
-            image = pipe(prompt=prompt, generator=generator, guidance_scale=7.5, num_inference_steps=30).images[0]
-            return image, seed
-
-        prompt = gr.Textbox(label="Prompt")
-        seed = gr.Slider(0, MAX_SEED, label="Seed", step=1, value=0)
-        generate_button = gr.Button("Generate")
-        result_image = gr.Image(label="Generated Image")
-
-        generate_button.click(
-            preloaded_infer,
-            inputs=[prompt, seed],
-            outputs=[result_image, seed]
-        )
+        gr.Markdown("## Works well on CPU and it is faster.")
+        preloaded_demo = gr.load("models/ZB-Tech/Text-to-Image")
 
     return preloaded_demo
 
 
 # Combine both interfaces in tabs
 with gr.Blocks() as demo:
-    with gr.Tab("Quick Image Generation"):
+    with gr.Tab("Faster Image Generation"):
         preloaded_ui = preloaded_model_ui()
 
-    with gr.Tab("Advanced Image Generation"):
+    with gr.Tab("Customizable Image Generation"):
         custom_ui = custom_model_ui()
 
 if __name__ == "__main__":