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("## Custom Model: Stability AI SDXL")
+        gr.Markdown("##Needs a GPU for best performance. Highly customizable for advanced users.")
         with gr.Row():
             prompt = gr.Text(label="Prompt")
             run_button = gr.Button("Generate")
@@ -67,18 +67,18 @@ def custom_model_ui():
 # Preloaded Gradio model
 def preloaded_model_ui():
     with gr.Blocks() as preloaded_demo:
-        gr.Markdown("## Preloaded Model: ZB-Tech Text-to-Image")
-        preloaded_demo = gr.load("models/ZB-Tech/Text-to-Image")
+        gr.Markdown("## Works well on CPU or GPU. Quick and ready-to-use for text-to-image generation.")
+        #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("Preloaded Model"):
+    with gr.Tab("Faster Image Genaration"):
         preloaded_ui = preloaded_model_ui()
 
-    with gr.Tab("Custom Model"):
+    with gr.Tab("Slower but More Customizable Image Generation"):
         custom_ui = custom_model_ui()
 
 if __name__ == "__main__":