Sourudra commited on
Commit
363fcc4
·
verified ·
1 Parent(s): 70bf4c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -41,7 +41,7 @@ def custom_infer(
41
  # Gradio interface for custom model
42
  def custom_model_ui():
43
  with gr.Blocks() as custom_demo:
44
- gr.Markdown("## Needs a GPU for best performance. Highly customizable for advanced users.")
45
  with gr.Row():
46
  prompt = gr.Text(label="Prompt")
47
  run_button = gr.Button("Generate")
@@ -67,7 +67,7 @@ def custom_model_ui():
67
  # Preloaded Gradio model
68
  def preloaded_model_ui():
69
  with gr.Blocks() as preloaded_demo:
70
- gr.Markdown("## Works well on CPU or GPU. Quick and ready-to-use for text-to-image generation.")
71
  preloaded_demo = gr.load("models/ZB-Tech/Text-to-Image")
72
 
73
  return preloaded_demo
@@ -75,10 +75,10 @@ def preloaded_model_ui():
75
 
76
  # Combine both interfaces in tabs
77
  with gr.Blocks() as demo:
78
- with gr.Tab("Faster Image Generation"):
79
  preloaded_ui = preloaded_model_ui()
80
 
81
- with gr.Tab("Customizable Image Generation"):
82
  custom_ui = custom_model_ui()
83
 
84
  if __name__ == "__main__":
 
41
  # Gradio interface for custom model
42
  def custom_model_ui():
43
  with gr.Blocks() as custom_demo:
44
+ gr.Markdown("## Advanced Text-to-Image Generation")
45
  with gr.Row():
46
  prompt = gr.Text(label="Prompt")
47
  run_button = gr.Button("Generate")
 
67
  # Preloaded Gradio model
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
 
75
 
76
  # Combine both interfaces in tabs
77
  with gr.Blocks() as demo:
78
+ with gr.Tab("Quick Image Generation"):
79
  preloaded_ui = preloaded_model_ui()
80
 
81
+ with gr.Tab("Advanced Image Generation"):
82
  custom_ui = custom_model_ui()
83
 
84
  if __name__ == "__main__":