Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("##
|
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("##
|
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("
|
79 |
preloaded_ui = preloaded_model_ui()
|
80 |
|
81 |
-
with gr.Tab("
|
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__":
|