Spaces:
Runtime error
Runtime error
omerbartal
commited on
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.")
|