Research-EAI commited on
Commit
93bf4fb
·
verified ·
1 Parent(s): 439b69a

added spaces.gpu

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -17,6 +17,7 @@ MAX_INPUT_TOKEN_LENGTH = 8192
17
 
18
  client = Together()
19
 
 
20
  def chat_fn(message, history, system_message, max_tokens, temperature, top_p):
21
 
22
  # Add system prompt
 
17
 
18
  client = Together()
19
 
20
+ @spaces.GPU
21
  def chat_fn(message, history, system_message, max_tokens, temperature, top_p):
22
 
23
  # Add system prompt