efederici commited on
Commit
ba5d9a3
1 Parent(s): 8d5bd81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ else:
45
  model = model.to(device)
46
 
47
 
48
- @spaces.GPU(duration=60)
49
  def chat(message, history, system, temperature, do_sample, max_tokens):
50
  chat = [{"role": "system", "content": system}] if system else []
51
  chat.extend(
 
45
  model = model.to(device)
46
 
47
 
48
+ @spaces.GPU()
49
  def chat(message, history, system, temperature, do_sample, max_tokens):
50
  chat = [{"role": "system", "content": system}] if system else []
51
  chat.extend(