Spaces:
Running
on
A100
Running
on
A100
Avijit Ghosh
commited on
Commit
•
396f720
1
Parent(s):
4e76f82
reduce time to zerogpu
Browse files
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=
|
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)
|