ayush2607 commited on
Commit
b69173f
·
verified ·
1 Parent(s): 150fc7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def process_file_streaming(img, model, processor):
51
 
52
  # Stream tokens
53
  streamer = TextIteratorStreamer(processor.tokenizer, skip_prompt=True, skip_special_tokens=True)
54
- generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=200)
55
 
56
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
57
  thread.start()
 
51
 
52
  # Stream tokens
53
  streamer = TextIteratorStreamer(processor.tokenizer, skip_prompt=True, skip_special_tokens=True)
54
+ generation_kwargs = dict(inputs, streamer=streamer, max_new_tokens=600)
55
 
56
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
57
  thread.start()