Update template/app.py

template/app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 from PIL import Image
 
-model_name = '$model_name'
+model_id = '$model_id'
 prefix = '$prefix'
      
 scheduler = DPMSolverMultistepScheduler(
@@ -20,12 +20,12 @@ scheduler = DPMSolverMultistepScheduler(
 )
 
 pipe = StableDiffusionPipeline.from_pretrained(
-  model_name,
+  model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler)
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
-  model_name,
+  model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler)
 
@@ -94,11 +94,10 @@ with gr.Blocks(css=css) as demo:
         f"""
             <div class="main-div">
               <div>
-                <h1>$model_name</h1>
+                <h1>$title</h1>
               </div>
               <p>
-               Demo for the <a href="https://huggingface.co/$model_name">$model_name</a> Stable Diffusion model.<br>
-               Add the following tokens to your prompts for the effect: <b>$prefix</b>.
+               $description
               </p>
               Running on <b>{"GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"}</b>
             </div>