Update app.py

app.py

@@ -46,15 +46,13 @@ def infer(
         'generator': generator,
     }
 
-    if model_id != model_repo_id_default:
+    if model_id != model_id_default:
         pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
         pipe = pipe.to(device)
         image = pipe(**params).images[0]
     else:
         image = pipe_default(**params).images[0]
 
-    
-
     return image, pipe.name_or_path
 
 css = """