Spaces:
Running
on
Zero
Running
on
Zero
app.py
CHANGED
|
@@ -140,7 +140,7 @@ def create_messages(review_fields, paper_text):
|
|
| 140 |
]
|
| 141 |
return messages
|
| 142 |
|
| 143 |
-
@spaces.GPU(
|
| 144 |
def convert_file(filepath):
|
| 145 |
full_text, images, out_metadata = convert_single_pdf(
|
| 146 |
filepath, model_refs, metadata=metadata, max_pages=MAX_PAGES
|
|
@@ -164,7 +164,7 @@ def process_file(file):
|
|
| 164 |
raise ValueError()
|
| 165 |
except spaces.zero.gradio.HTMLError as e:
|
| 166 |
print(e)
|
| 167 |
-
return "GPU quota exceeded"
|
| 168 |
except Exception as e:
|
| 169 |
print(traceback.format_exc())
|
| 170 |
print(f"Error converting {filepath}: {e}")
|
|
@@ -172,7 +172,7 @@ def process_file(file):
|
|
| 172 |
return paper_text
|
| 173 |
|
| 174 |
|
| 175 |
-
@spaces.GPU(duration=
|
| 176 |
def generate(paper_text, review_template):
|
| 177 |
# messages = [
|
| 178 |
# {"role": "system", "content": "You are a pirate."},
|
|
@@ -225,14 +225,15 @@ with gr.Blocks(theme=theme, css=css) as demo:
|
|
| 225 |
description = gr.Markdown(description)
|
| 226 |
instr = gr.Markdown("## Upload your paper in pdf format")
|
| 227 |
file_input = gr.File(file_types=[".pdf"], file_count="single")
|
| 228 |
-
paper_text_field= gr.Textbox(label="Paper Text", max_lines=20, autoscroll=False)
|
| 229 |
-
|
|
|
|
| 230 |
generate_button = gr.Button("Generate Review", interactive=not paper_text_field)
|
| 231 |
# generate_button = gr.Button("Generate Review")
|
| 232 |
file_input.upload(process_file, file_input, paper_text_field)
|
| 233 |
paper_text_field.change(lambda text: gr.update(interactive=True) if len(text) > 200 else gr.update(interactive=False), paper_text_field, generate_button)
|
| 234 |
|
| 235 |
-
review_field = gr.Markdown("# Review", label="Review")
|
| 236 |
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)
|
| 237 |
demo.title = "Paper Review Generator"
|
| 238 |
|
|
|
|
| 140 |
]
|
| 141 |
return messages
|
| 142 |
|
| 143 |
+
@spaces.GPU()
|
| 144 |
def convert_file(filepath):
|
| 145 |
full_text, images, out_metadata = convert_single_pdf(
|
| 146 |
filepath, model_refs, metadata=metadata, max_pages=MAX_PAGES
|
|
|
|
| 164 |
raise ValueError()
|
| 165 |
except spaces.zero.gradio.HTMLError as e:
|
| 166 |
print(e)
|
| 167 |
+
return "Error. GPU quota exceeded. Please return later."
|
| 168 |
except Exception as e:
|
| 169 |
print(traceback.format_exc())
|
| 170 |
print(f"Error converting {filepath}: {e}")
|
|
|
|
| 172 |
return paper_text
|
| 173 |
|
| 174 |
|
| 175 |
+
@spaces.GPU(duration=120)
|
| 176 |
def generate(paper_text, review_template):
|
| 177 |
# messages = [
|
| 178 |
# {"role": "system", "content": "You are a pirate."},
|
|
|
|
| 225 |
description = gr.Markdown(description)
|
| 226 |
instr = gr.Markdown("## Upload your paper in pdf format")
|
| 227 |
file_input = gr.File(file_types=[".pdf"], file_count="single")
|
| 228 |
+
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)
|
| 229 |
+
with gr.Accordion("Show Review Template"):
|
| 230 |
+
review_template_field = gr.Textbox(label="Review Template", max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
|
| 231 |
generate_button = gr.Button("Generate Review", interactive=not paper_text_field)
|
| 232 |
# generate_button = gr.Button("Generate Review")
|
| 233 |
file_input.upload(process_file, file_input, paper_text_field)
|
| 234 |
paper_text_field.change(lambda text: gr.update(interactive=True) if len(text) > 200 else gr.update(interactive=False), paper_text_field, generate_button)
|
| 235 |
|
| 236 |
+
review_field = gr.Markdown("# Review\n\n\n", label="Review")
|
| 237 |
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)
|
| 238 |
demo.title = "Paper Review Generator"
|
| 239 |
|