nxphi47 commited on
Commit
46573d8
1 Parent(s): 1157296

Update multipurpose_chatbot/demos/text_completion.py

Browse files
multipurpose_chatbot/demos/text_completion.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  from gradio.themes import ThemeClass as Theme
3
  import numpy as np
@@ -61,7 +62,7 @@ from ..configs import (
61
 
62
  from ..globals import MODEL_ENGINE
63
 
64
-
65
  def generate_text_completion_stream_engine(
66
  message: str,
67
  temperature: float,
 
1
+ import spaces
2
  import os
3
  from gradio.themes import ThemeClass as Theme
4
  import numpy as np
 
62
 
63
  from ..globals import MODEL_ENGINE
64
 
65
+ @spaces.GPU
66
  def generate_text_completion_stream_engine(
67
  message: str,
68
  temperature: float,