aixsatoshi commited on
Commit
133be07
1 Parent(s): da8a347

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ h3 {
41
  }
42
  """
43
 
44
- @gr.GPU
45
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
46
  print(f'Message: {message}')
47
  print(f'History: {history}')
 
41
  }
42
  """
43
 
44
+ @spaces.GPU(duration=120)
45
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
46
  print(f'Message: {message}')
47
  print(f'History: {history}')