a

app.py

@@ -26,7 +26,7 @@ model_refs = load_all_models()
 metadata = {}
 
 # prepare LLM
-model_name = "maxidl/arena-test"
+model_name = "maxidl/Llama-OpenReviewer-8B"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     torch_dtype=torch.bfloat16,
@@ -189,7 +189,8 @@ def generate(paper_text, review_template):
         input_ids=input_ids,
         max_new_tokens=2048,
         do_sample=True,
-        temperature=0.8,
+        temperature=0.6,
+        top_p=0.9
     )
     generated_ids = [
         output_ids[len(input_ids):] for input_ids, output_ids in zip(input_ids, generated_ids)
@@ -205,10 +206,10 @@ def generate(paper_text, review_template):
 css = """
 #warning {background: red;}
 
-.gradio-container {background-color: #FFFDFA !important;}
-.title {background-color: #FFFDFA !important;}
 .button {background-color: #8C1B13 !important; color: white !important;}
 """
+# .gradio-container {background-color: #FFFDFA !important;}
+# .title {background-color: #FFFDFA !important;}
 #8C1B13 red
 #4D8093 blue
 #767676 med grey
@@ -217,23 +218,42 @@ css = """
 #FFFDFA white
 
 title = "# OpenReviewer"
-description = """Placeholder Description"""
+title_icons = """<div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
+<a href="https://huggingface.co/maxidl/Llama-OpenReviewer-8B""><img src="https://img.shields.io/badge/🤗-Model-ffbd45.svg" alt="HuggingFace"></a>
+</div>
+"""
+
+description = """This is an online demo of the `Llama-OpenReviewer-8B` model, which generates high-quality reviews for machine learning and AI-related papers.
+
+## Demo Guidelines
+
+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.
+
+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.
+
+3. Having obtained a markdown version of your paper, you can now click `Generate Review`.
+
+4. (Optional) Take a look at the Review Template to properly interpret the generated review. You can also change the review template before generating in case you want to generate a review with a different schema and aspects.
+
+
+"""
 
-theme = gr.themes.Default()
+theme = gr.themes.Default(primary_hue="gray", secondary_hue="blue", neutral_hue="slate")
 with gr.Blocks(theme=theme, css=css) as demo:
     title = gr.Markdown(title, elem_classes=["title"])
+    title_icons = gr.Markdown(title_icons)
     description = gr.Markdown(description)
-    instr = gr.Markdown("## Upload your paper in pdf format")
+    instr = gr.Markdown("Upload your paper in pdf format")
     file_input = gr.File(file_types=[".pdf"], file_count="single")
     paper_text_field= gr.Textbox("Upload a pdf or paste the full text of your paper in markdown format here.", label="Paper Text", max_lines=20, autoscroll=False)
-    with gr.Accordion("Show Review Template"):
-        review_template_field = gr.Textbox(label="Review Template", max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
+    with gr.Accordion("Review Template", open=False):
+        review_template_description = gr.Markdown("We use the ICLR 2025 review template by default, but you can modify the template below as you like.")
+        review_template_field = gr.Textbox(max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
     generate_button = gr.Button("Generate Review", interactive=not paper_text_field)
-    # generate_button = gr.Button("Generate Review")
     file_input.upload(process_file, file_input, paper_text_field)
     paper_text_field.change(lambda text: gr.update(interactive=True) if len(text) > 200 else gr.update(interactive=False), paper_text_field, generate_button)
 
-    review_field = gr.Markdown("# Review\n\n\n", label="Review")
+    review_field = gr.Markdown("# Review\n\n\n\n\n", label="Review")
     generate_button.click(fn=lambda: gr.update(interactive=False), inputs=None, outputs=generate_button).then(generate, [paper_text_field, review_template_field], review_field).then(fn=lambda: gr.update(interactive=True), inputs=None, outputs=generate_button)
     demo.title = "Paper Review Generator"