Update app.py
Browse files
app.py
CHANGED
@@ -45,8 +45,8 @@ def generate_api():
|
|
45 |
# Extract required fields from the request
|
46 |
prompt = data.get('prompt', '')
|
47 |
negative_prompt = data.get('negative_prompt', None)
|
48 |
-
height = data.get('height',
|
49 |
-
width = data.get('width',
|
50 |
num_inference_steps = data.get('num_inference_steps', 50) # Default number of inference steps
|
51 |
guidance_scale = data.get('guidance_scale', 7.5) # Default guidance scale
|
52 |
model_name = data.get('model', 'stabilityai/stable-diffusion-2-1') # Default model
|
|
|
45 |
# Extract required fields from the request
|
46 |
prompt = data.get('prompt', '')
|
47 |
negative_prompt = data.get('negative_prompt', None)
|
48 |
+
height = data.get('height', 1024) # Default height
|
49 |
+
width = data.get('width', 720) # Default width
|
50 |
num_inference_steps = data.get('num_inference_steps', 50) # Default number of inference steps
|
51 |
guidance_scale = data.get('guidance_scale', 7.5) # Default guidance scale
|
52 |
model_name = data.get('model', 'stabilityai/stable-diffusion-2-1') # Default model
|