Avijit Ghosh commited on
Commit
396f720
1 Parent(s): 4e76f82

reduce time to zerogpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ def skintoneplot(hex_codes):
120
  ax.add_patch(plt.Rectangle((0, 0), 1, 1, color=sorted_hex_codes[i]))
121
  return fig
122
 
123
- @spaces.GPU(duration=400)
124
  def generate_images_plots(prompt, model_name):
125
  global pipeline_text2image
126
  pipeline_text2image = load_model(model_name)
 
120
  ax.add_patch(plt.Rectangle((0, 0), 1, 1, color=sorted_hex_codes[i]))
121
  return fig
122
 
123
+ @spaces.GPU(duration=300)
124
  def generate_images_plots(prompt, model_name):
125
  global pipeline_text2image
126
  pipeline_text2image = load_model(model_name)