John6666 commited on
Commit
1553d93
1 Parent(s): c60bd9d

Upload mod.py

Browse files
Files changed (1) hide show
  1. mod.py +1 -0
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']