Update app.py

A model_id parameter is added to infer function.

app.py

@@ -23,14 +23,15 @@ MAX_IMAGE_SIZE = 1024
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
-    prompt,
-    negative_prompt,
-    seed,
+    model_id: Optional[str] = "CompVis/stable-diffusion-v1-4",
+    prompt: str,
+    negative_prompt=gr.Text(…),
+    seed: Optional[int] = 42,
     randomize_seed,
     width,
     height,
-    guidance_scale,
-    num_inference_steps,
+    guidance_scale: Optional[float] = 7,
+    num_inference_steps: Optional[int] = 20,
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed: