Spaces:
Runtime error
Runtime error
Commit
·
6fd67ae
1
Parent(s):
4111a9c
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,6 @@ pipe = pipe.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
|
|
| 26 |
|
| 27 |
def generate_image_fn(prompt: str, img_width: int, img_height=512) -> list:
|
| 28 |
start_time = time.time()
|
| 29 |
-
prompt = "a photo of the dolomites"
|
| 30 |
image = pipe(prompt, height=img_height, width=img_width).images
|
| 31 |
end_time = time.time()
|
| 32 |
print(f"Time taken: {end_time - start_time} seconds.")
|
|
|
|
| 26 |
|
| 27 |
def generate_image_fn(prompt: str, img_width: int, img_height=512) -> list:
|
| 28 |
start_time = time.time()
|
|
|
|
| 29 |
image = pipe(prompt, height=img_height, width=img_width).images
|
| 30 |
end_time = time.time()
|
| 31 |
print(f"Time taken: {end_time - start_time} seconds.")
|