Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
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("## 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__":
|