maxidl commited on
Commit
244ed6d
·
1 Parent(s): bdd2cf9
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -140,7 +140,7 @@ def create_messages(review_fields, paper_text):
140
  ]
141
  return messages
142
 
143
- @spaces.GPU(duration=60)
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=60)
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
- review_template_field = gr.Textbox(label="Review Template", max_lines=20, autoscroll=False, value=REVIEW_FIELDS)
 
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