Update app.py

app.py

@@ -7,7 +7,8 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
+# model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
+model_repo_id = "CompVis/stable-diffusion-v1-4"
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16
@@ -21,17 +22,17 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 
-# @spaces.GPU #[uncomment to use ZeroGPU]
+@spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
-    model_id: Optional[str] = "CompVis/stable-diffusion-v1-4",
+    # model_id: Optional[str] = "CompVis/stable-diffusion-v1-4",
     negative_prompt=gr.Text('cat, dog'),
-    prompt: str,
-    seed,
+    prompt: str = 'cute_animal',
+    seed: Optional[int] = 42,
     randomize_seed,
-    width,
-    height,
-    guidance_scale,
-    num_inference_steps,
+    width = 128,
+    height = 128,
+    guidance_scale: Optional[float] = 7,
+    num_inference_steps: Optional[int] = 20,
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed: