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 |
|