Update multipurpose_chatbot/demos/chat_interface.py

multipurpose_chatbot/demos/chat_interface.py

@@ -1,4 +1,5 @@
 import os
+import spaces
 from gradio.themes import ThemeClass as Theme
 import numpy as np
 import argparse
@@ -120,7 +121,8 @@ def format_conversation(history, system_prompt=None):
         _str = f"<<<Syst>>> {system_prompt}\n" + _str
     return _str
 
-    
+
+@spaces.GPU
 def chat_response_stream_multiturn_engine(
     message: str, 
     history: List[Tuple[str, str]],