Rijgersberg commited on
Commit
e0db9a1
1 Parent(s): 123413a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
10
  model = AutoModelForCausalLM.from_pretrained(model_id).to("cuda")
11
 
12
 
13
- @spaces.GPU(duration=30)
14
  def generate(
15
  message,
16
  history: list[tuple[str, str]],
 
10
  model = AutoModelForCausalLM.from_pretrained(model_id).to("cuda")
11
 
12
 
13
+ @spaces.GPU
14
  def generate(
15
  message,
16
  history: list[tuple[str, str]],