Spaces:
Running
on
A100
Running
on
A100
Avijit Ghosh
commited on
Commit
•
3245b5c
1
Parent(s):
396f720
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=200)
|
124 |
def generate_images_plots(prompt, model_name):
|
125 |
global pipeline_text2image
|
126 |
pipeline_text2image = load_model(model_name)
|