Sourudra commited on
Commit
f088c34
·
verified ·
1 Parent(s): ce2c089

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -41,7 +41,8 @@ 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 and it is highly customizable.")
 
45
  with gr.Row():
46
  prompt = gr.Text(label="Prompt")
47
  run_button = gr.Button("Generate")
@@ -75,10 +76,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("## Needs a GPU for best performance and it is highly customizable.\n stabilityai/sdxl-turbo")
45
+
46
  with gr.Row():
47
  prompt = gr.Text(label="Prompt")
48
  run_button = gr.Button("Generate")
 
76
 
77
  # Combine both interfaces in tabs
78
  with gr.Blocks() as demo:
79
+ with gr.Tab("Quick Image Generation"):
80
  preloaded_ui = preloaded_model_ui()
81
 
82
+ with gr.Tab("Advanced Image Generation"):
83
  custom_ui = custom_model_ui()
84
 
85
  if __name__ == "__main__":