Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 from threading import Thread
 from typing import Iterator
 
@@ -29,7 +30,7 @@ LICENSE = """
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU  This demo does not work on CPU.</p>"
 
-
+@spaces.GPU(duration=120)
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],