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("##Needs a GPU for best performance. Highly customizable for advanced users.")
+        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")
@@ -68,14 +68,14 @@ def custom_model_ui():
 def preloaded_model_ui():
     with gr.Blocks() as preloaded_demo:
         gr.Markdown("## Works well on CPU or GPU. Quick and ready-to-use for text-to-image generation.")
-        preloaded_demo = gr.load()
+        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("Faster Image Genaration"):
+    with gr.Tab("Faster Image Generation"):
         preloaded_ui = preloaded_model_ui()
 
     with gr.Tab("Customizable Image Generation"):