Spaces:
Running
on
Zero
Running
on
Zero
Upload mod.py
Browse files
mod.py
CHANGED
@@ -167,6 +167,7 @@ def load_prompt_enhancer():
|
|
167 |
enhancer_flux = load_prompt_enhancer()
|
168 |
|
169 |
|
|
|
170 |
def enhance_prompt(input_prompt):
|
171 |
result = enhancer_flux("enhance prompt: " + input_prompt, max_length = 256)
|
172 |
enhanced_text = result[0]['generated_text']
|
|
|
167 |
enhancer_flux = load_prompt_enhancer()
|
168 |
|
169 |
|
170 |
+
@spaces.GPU(duration=30)
|
171 |
def enhance_prompt(input_prompt):
|
172 |
result = enhancer_flux("enhance prompt: " + input_prompt, max_length = 256)
|
173 |
enhanced_text = result[0]['generated_text']
|