Update multipurpose_chatbot/demos/multimodal_chat_interface.py

multipurpose_chatbot/demos/multimodal_chat_interface.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 from gradio.themes import ThemeClass as Theme
 import numpy as np
@@ -903,6 +904,7 @@ def gradio_history_to_vision_doc_conversation_prompt_paths(
     return full_prompt, image_paths, conversations
 
 
+@spaces.GPU
 def vision_chat_response_stream_multiturn_engine(
     history: List[Tuple[str, str]], 
     temperature: float, 
@@ -955,6 +957,7 @@ def vision_chat_response_stream_multiturn_engine(
         yield response, num_tokens
 
 
+@spaces.GPU
 def doc_chat_response_stream_multiturn_engine(
     history: List[Tuple[str, str]], 
     temperature: float, 
@@ -1005,7 +1008,7 @@ def doc_chat_response_stream_multiturn_engine(
 
 
 
-
+@spaces.GPU
 def vision_doc_chat_response_stream_multiturn_engine(
     history: List[Tuple[str, str]], 
     temperature: float,