MohamedRashad commited on
Commit
641afde
1 Parent(s): 457f149

chore: Update GPU duration in img2img_infer function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ Based on what i gave you, Write a great short AI Art Prompt for me that is based
65
  print(f"{Fore.GREEN}{refined_prompt}{Style.RESET_ALL}")
66
  return refined_prompt
67
 
68
- @spaces.GPU
69
  def img2img_infer(image_path, image_description):
70
  pil_image = Image.open(image_path)
71
  width, height = pil_image.size
 
65
  print(f"{Fore.GREEN}{refined_prompt}{Style.RESET_ALL}")
66
  return refined_prompt
67
 
68
+ @spaces.GPU(duration=75)
69
  def img2img_infer(image_path, image_description):
70
  pil_image = Image.open(image_path)
71
  width, height = pil_image.size