Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import spaces
 import threading
 import time
 import subprocess
@@ -52,6 +53,8 @@ h3 {
     text-align: center;
 }
 """
+
+@spaces.GPU()
 def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
     print(f'message is - {message}')
     print(f'history is - {history}')