maxidl commited on
Commit
b451083
·
1 Parent(s): 48d6ef4
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -187,7 +187,7 @@ def generate(paper_text, review_template):
187
  print(f"input_ids shape: {input_ids.shape}")
188
  generated_ids = model.generate(
189
  input_ids=input_ids,
190
- max_new_tokens=2048,
191
  do_sample=True,
192
  temperature=0.6,
193
  top_p=0.9
@@ -218,7 +218,7 @@ description = """This is an online demo featuring [Llama-OpenReviewer-8B](https:
218
 
219
  ## Demo Guidelines
220
 
221
- 1. Upload you paper as a pdf file. Alternatively you can paste the full text of your paper in markdown format below. Note that we do **not** store anything. All data is kept in ephemeral storage during processing.
222
 
223
  2. Once you upload a pdf it will be converted to markdown. This takes some time as it runs multiple transformer models to parse the layout and extract text and tables. Checkout [marker](https://github.com/VikParuchuri/marker/tree/master) for details.
224
 
@@ -241,7 +241,7 @@ with gr.Blocks(theme=theme) as demo:
241
  paper_text_field= gr.Textbox("Upload a pdf or paste the full text of your paper in markdown format here.", label="Paper Text", lines=20, max_lines=20, autoscroll=False)
242
  with gr.Accordion("Review Template", open=False):
243
  review_template_description = gr.Markdown("We use the ICLR 2025 review template by default, but you can modify the template below as you like.")
244
- review_template_field = gr.Textbox(min_lines=20, max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
245
  generate_button = gr.Button("Generate Review", interactive=not paper_text_field)
246
  file_input.upload(process_file, file_input, paper_text_field)
247
  paper_text_field.change(lambda text: gr.update(interactive=True) if len(text) > 200 else gr.update(interactive=False), paper_text_field, generate_button)
 
187
  print(f"input_ids shape: {input_ids.shape}")
188
  generated_ids = model.generate(
189
  input_ids=input_ids,
190
+ max_new_tokens=4096,
191
  do_sample=True,
192
  temperature=0.6,
193
  top_p=0.9
 
218
 
219
  ## Demo Guidelines
220
 
221
+ 1. Upload you paper as a pdf file. Alternatively you can paste the full text of your paper in markdown format below. We do **not** store your data. User data is kept in ephemeral storage during processing.
222
 
223
  2. Once you upload a pdf it will be converted to markdown. This takes some time as it runs multiple transformer models to parse the layout and extract text and tables. Checkout [marker](https://github.com/VikParuchuri/marker/tree/master) for details.
224
 
 
241
  paper_text_field= gr.Textbox("Upload a pdf or paste the full text of your paper in markdown format here.", label="Paper Text", lines=20, max_lines=20, autoscroll=False)
242
  with gr.Accordion("Review Template", open=False):
243
  review_template_description = gr.Markdown("We use the ICLR 2025 review template by default, but you can modify the template below as you like.")
244
+ review_template_field = gr.Textbox(lines=20, max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
245
  generate_button = gr.Button("Generate Review", interactive=not paper_text_field)
246
  file_input.upload(process_file, file_input, paper_text_field)
247
  paper_text_field.change(lambda text: gr.update(interactive=True) if len(text) > 200 else gr.update(interactive=False), paper_text_field, generate_button)