Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ pipe = load_lora_model(pipe, lora_model_path)
|
|
| 36 |
|
| 37 |
# Function to generate an image from a text prompt
|
| 38 |
def generate_image(prompt):
|
| 39 |
-
image = pipe(prompt).images[0]
|
| 40 |
return image
|
| 41 |
|
| 42 |
# Gradio interface
|
|
|
|
| 36 |
|
| 37 |
# Function to generate an image from a text prompt
|
| 38 |
def generate_image(prompt):
|
| 39 |
+
image = pipe(prompt, height=1080, width=1080).images[0]
|
| 40 |
return image
|
| 41 |
|
| 42 |
# Gradio interface
|