Upload mod.py

mod.py

@@ -167,6 +167,7 @@ def load_prompt_enhancer():
 enhancer_flux = load_prompt_enhancer()
 
 
+@spaces.GPU(duration=30)
 def enhance_prompt(input_prompt):
     result = enhancer_flux("enhance prompt: " + input_prompt, max_length = 256)
     enhanced_text = result[0]['generated_text']