toshi456 commited on
Commit
99a04ab
1 Parent(s): dfb237c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ model = LlavaGpt2ForCausalLM.from_pretrained(
23
  )
24
  tokenizer = transformers.AutoTokenizer.from_pretrained(
25
  model_path,
26
- model_max_length=1024,
27
  padding_side="right",
28
  use_fast=False,
29
  )
@@ -179,4 +179,4 @@ with gr.Blocks() as demo:
179
 
180
 
181
  if __name__ == "__main__":
182
- demo.queue().launch(share=True)
 
23
  )
24
  tokenizer = transformers.AutoTokenizer.from_pretrained(
25
  model_path,
26
+ model_max_length=1532,
27
  padding_side="right",
28
  use_fast=False,
29
  )
 
179
 
180
 
181
  if __name__ == "__main__":
182
+ demo.queue().launch()