Sergidev commited on
Commit
2911cde
Β·
verified Β·
1 Parent(s): c2c6f60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ model = AutoModelForCausalLM.from_pretrained(
28
  )
29
 
30
  model.eval()
31
- @spaces.GPU(duration=90)
32
  def generate(
33
  message: str,
34
  max_new_tokens: int = 1024,
 
28
  )
29
 
30
  model.eval()
31
+ @spaces.GPU
32
  def generate(
33
  message: str,
34
  max_new_tokens: int = 1024,