omerbartal commited on
Commit
6fd67ae
1 Parent(s): 4111a9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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.")