Update app.py

app.py

@@ -12,12 +12,14 @@ from transformers import (
     TextIteratorStreamer,
 )
 
-DESCRIPTION = """\
+GENERATION_TIME=90
+
+DESCRIPTION = f"""\
 # ORLM LLaMA-3-8B
 
 Hello! I'm ORLM-LLaMA-3-8B, here to automate your optimization modeling tasks! Check our [repo](https://github.com/Cardinal-Operations/ORLM) and [paper](https://arxiv.org/abs/2405.17743)!
 
-Please note that solution generation may be terminated if it exceeds 100 seconds. We strongly recommend running the demo locally using our [sample script](https://github.com/Cardinal-Operations/ORLM/blob/master/scripts/inference.py) for a smoother experience.
+Please note that solution generation may be terminated if it exceeds {GENERATION_TIME} seconds. We strongly recommend running the demo locally using our [sample script](https://github.com/Cardinal-Operations/ORLM/blob/master/scripts/inference.py) for a smoother experience.
 
 If the demo successfully generates a code solution, execute it in your Python environment with `coptpy` installed to obtain the final optimal value for your task.
 """
@@ -47,7 +49,7 @@ Below is an operations research question. Build a mathematical model and corresp
 # Response:
 """
 
-@spaces.GPU(duration=90)
+@spaces.GPU(duration=GENERATION_TIME)
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],