ghengx commited on
Commit
df2e419
1 Parent(s): ae40c8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model = AutoModelForCausalLM.from_pretrained(
15
  )
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
 
18
- streamer = TextIteratorStreamer(tokenizer, timeout=100., skip_prompt=True, skip_special_tokens=True)
19
 
20
  @spaces.GPU
21
  def respond(
 
15
  )
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
 
18
+ streamer = TextIteratorStreamer(tokenizer, timeout=300, skip_prompt=True, skip_special_tokens=True)
19
 
20
  @spaces.GPU
21
  def respond(