a

app.py

@@ -21,7 +21,9 @@ MAX_PAGES = 20
 MIN_LENGTH=200
 settings.EXTRACT_IMAGES = False
 settings.DEBUG = False
-settings.PDFTEXT_CPU_WORKERS=1
+settings.PDFTEXT_CPU_WORKERS = 1
+settings.DETECTOR_POSTPROCESSING_CPU_WORKERS = 1
+settings.OCR_PARALLEL_WORKERS = 1
 surya_settings.IN_STREAMLIT = True
 
 model_refs = load_all_models()
@@ -177,10 +179,6 @@ def process_file(file):
 
 @spaces.GPU(duration=90)
 def generate(paper_text, review_template):
-    # messages = [
-    #     {"role": "system", "content": "You are a pirate."},
-    #     {"role": "user", "content": paper_text}
-    # ]
     messages = create_messages(review_template, paper_text)
     input_ids = tokenizer.apply_chat_template(
         messages,
@@ -195,20 +193,6 @@ def generate(paper_text, review_template):
     for new_text in streamer:
         generated_text += new_text
         yield generated_text.replace("<|eot_id|>", "")
-    # generated_ids = model.generate(
-    #     input_ids=input_ids,
-    #     max_new_tokens=4096,
-    #     do_sample=True,
-    #     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)
-    # ]
-    
-    # response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    # return response
-    # return "Success"
 
 
 
@@ -246,8 +230,6 @@ theme = gr.themes.Default(primary_hue="gray", secondary_hue="blue", neutral_hue=
 with gr.Blocks(theme=theme) as demo:
     title = gr.HTML(title)
     description = gr.Markdown(description)
-    # login_button = gr.LoginButton()
-    # 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", lines=20, max_lines=20, autoscroll=False)
     with gr.Accordion("Review Template", open=False):
@@ -259,14 +241,6 @@ with gr.Blocks(theme=theme) as demo:
 
     review_field = gr.Markdown("\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)
-    
-    # gr.Examples([
-    #     ["examples/2105.04505v1.pdf", REVIEW_FIELDS]
-    # ],
-    # inputs=[paper_text_field, review_template_field],
-    # outputs=[review_field],
-    # fn=generate,
-    # cache_examples=True)
 
     demo.title = "OpenReviewer"