:memo: Change button location

app.py

@@ -5,10 +5,14 @@ from PIL import Image
 from diffusers import DiffusionPipeline
 
 DIFFUSERS_MODEL_IDS = [
+    # SD Models
     "stabilityai/stable-diffusion-3-medium-diffusers",
     "stabilityai/stable-diffusion-xl-base-1.0",
     "stabilityai/stable-diffusion-2-1",
     "runwayml/stable-diffusion-v1-5",
+
+    # Other Models
+    "Prgckwb/trpfrog-diffusion",
 ]
 
 EXTERNAL_MODEL_MAPPING = {
@@ -54,6 +58,9 @@ def inference(
         except Exception as e:
             raise gr.Error(str(e))
 
+    if device != 'cuda':
+        raise gr.Error("This model requires a GPU to run. Please switch to a GPU runtime.")
+
     image = pipe(
         prompt,
         negative_prompt=negative_prompt,
@@ -95,6 +102,9 @@ if __name__ == "__main__":
                     gr.Image(label="Image", type="pil"),
                 ]
 
+        btn = gr.Button("Generate")
+        btn.click(fn=inference, inputs=inputs, outputs=outputs)
+
         gr.Examples(
             examples=[
                 ['stabilityai/stable-diffusion-3-medium-diffusers', 'A cat holding a sign that says Hello world', ""],
@@ -104,7 +114,4 @@ if __name__ == "__main__":
             inputs=inputs,
         )
 
-        btn = gr.Button("Generate")
-        btn.click(fn=inference, inputs=inputs, outputs=outputs)
-
     demo.queue().launch()