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']
|