:tada: add external model

app.py

@@ -1,4 +1,7 @@
+import dataclasses
+
 import gradio as gr
+import requests
 import spaces
 import torch
 from PIL import Image
@@ -33,11 +36,39 @@ if device == 'cuda':
     ).to(device)
 
 
+@dataclasses.dataclass
+class Input:
+    prompt: str
+    model_id: str = "stabilityai/stable-diffusion-3-medium-diffusers"
+    negative_prompt: str = ''
+    width: int = 1024
+    height: int = 1024
+    guidance_scale: float = 7.5
+    num_inference_step: int = 28
+    num_images: int = 4
+
+    def to_list(self):
+        return [
+            self.prompt, self.model_id, self.negative_prompt,
+            self.width, self.height, self.guidance_scale,
+            self.num_inference_step, self.num_images
+        ]
+
+
+EXAMPLES = [
+    Input(prompt='A cat holding a sign that says Hello world').to_list(),
+    Input(
+        prompt='Beautiful pixel art of a Wizard with hovering text "Achivement unlocked: Diffusion models can spell now"'
+    ).to_list(),
+    Input(prompt='A corgi wearing sunglasses says "U-Net is OVER!!"').to_list(),
+]
+
+
 @spaces.GPU()
 @torch.inference_mode()
 def inference(
-        model_id: str,
         prompt: str,
+        model_id: str = "stabilityai/stable-diffusion-3-medium-diffusers",
         negative_prompt: str = "",
         width: int = 512,
         height: int = 512,
@@ -86,19 +117,22 @@ def inference(
 
 
 if __name__ == "__main__":
-    with gr.Blocks() as demo:
+    theme = gr.themes.Default(primary_hue=gr.themes.colors.emerald)
+
+    with gr.Blocks(theme=theme) as demo:
         gr.Markdown(f"# Stable Diffusion Demo")
 
         with gr.Row():
             with gr.Column():
+                prompt = gr.Text(label="Prompt", placeholder="Enter a prompt here")
+
                 model_id = gr.Dropdown(
                     label="Model ID",
                     choices=MODEL_CHOICES,
                     value="stabilityai/stable-diffusion-3-medium-diffusers",
                 )
-                prompt = gr.Text(label="Prompt", value="")
 
-                with gr.Accordion("Additional Settings (W.I.P)", open=False):
+                with gr.Accordion("Additional Settings", open=False):
                     negative_prompt = gr.Text(label="Negative Prompt", value="")
 
                     with gr.Row():
@@ -114,8 +148,8 @@ if __name__ == "__main__":
                 output_image = gr.Image(label="Image", type="pil")
 
         inputs = [
-            model_id,
             prompt,
+            model_id,
             negative_prompt,
             width,
             height,
@@ -132,14 +166,7 @@ if __name__ == "__main__":
         )
 
         gr.Examples(
-            examples=[
-                ['stabilityai/stable-diffusion-3-medium-diffusers', 'A cat holding a sign that says Hello world', ""],
-                ['stabilityai/stable-diffusion-3-medium-diffusers',
-                 'Beautiful pixel art of a Wizard with hovering text "Achivement unlocked: Diffusion models can spell now"',
-                 ''],
-                ['stabilityai/stable-diffusion-3-medium-diffusers', 'A corgi wearing sunglasses says "U-Net is OVER!!"',
-                 ''],
-            ],
+            examples=EXAMPLES,
             inputs=inputs,
         )
 

example.csv

@@ -1,4 +0,0 @@
-model_uri,description,additional_info
-stabilityai/stable-diffusion-3-medium-diffusers,"A cat holding a sign that says Hello world",
-stabilityai/stable-diffusion-3-medium-diffusers,"Beautiful pixel art of a Wizard with hovering text "Achivement unlocked: Diffusion models can spell now"",
-stabilityai/stable-diffusion-3-medium-diffusers,"A corgi wearing sunglasses says ""U-Net is OVER!!"",