1inkusFace commited on
Commit
4d756fd
·
verified ·
1 Parent(s): fce4d5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -240,7 +240,7 @@ def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
240
  f.write(f"Model UNET: realistic_porn \n")
241
  upload_to_ftp(filename)
242
 
243
- @spaces.GPU(duration=40)
244
  def generate_30(
245
  model_choice: str,
246
  prompt: str,
@@ -299,7 +299,7 @@ def generate_30(
299
  image_paths = [save_image(downscale1)]
300
  return image_paths
301
 
302
- @spaces.GPU(duration=60)
303
  def generate_60(
304
  model_choice: str,
305
  prompt: str,
@@ -355,7 +355,7 @@ def generate_60(
355
  image_paths = [save_image(downscale1)]
356
  return image_paths
357
 
358
- @spaces.GPU(duration=90)
359
  def generate_90(
360
  model_choice: str,
361
  prompt: str,
@@ -449,7 +449,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
449
  placeholder="Enter your prompt",
450
  container=False,
451
  )
452
- run_button_30 = gr.Button("Run 30 Seconds", scale=0)
453
  run_button_60 = gr.Button("Run 60 Seconds", scale=0)
454
  run_button_90 = gr.Button("Run 90 Seconds", scale=0)
455
  result = gr.Gallery(label="Result", columns=1, show_label=False)
 
240
  f.write(f"Model UNET: realistic_porn \n")
241
  upload_to_ftp(filename)
242
 
243
+ @spaces.GPU(duration=55)
244
  def generate_30(
245
  model_choice: str,
246
  prompt: str,
 
299
  image_paths = [save_image(downscale1)]
300
  return image_paths
301
 
302
+ @spaces.GPU(duration=70)
303
  def generate_60(
304
  model_choice: str,
305
  prompt: str,
 
355
  image_paths = [save_image(downscale1)]
356
  return image_paths
357
 
358
+ @spaces.GPU(duration=100)
359
  def generate_90(
360
  model_choice: str,
361
  prompt: str,
 
449
  placeholder="Enter your prompt",
450
  container=False,
451
  )
452
+ run_button_30 = gr.Button("Run 45 Seconds", scale=0)
453
  run_button_60 = gr.Button("Run 60 Seconds", scale=0)
454
  run_button_90 = gr.Button("Run 90 Seconds", scale=0)
455
  result = gr.Gallery(label="Result", columns=1, show_label=False)