Zaixi commited on
Commit
d45cade
·
1 Parent(s): fe8f637
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -430,11 +430,12 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
430
  with gr.Row():
431
  view3d = Molecule3D(label="3D Visualization", reps=reps)
432
  legend = gr.Markdown("""
433
- **Color Legend:**
 
 
 
 
434
 
435
- - <span style="color:grey">Unwatermarked Structure</span>
436
- - <span style="color:cyan">Watermarked Structure</span>
437
- """)
438
  with gr.Row():
439
  cif_file = gr.File(label="Download CIF File")
440
  with gr.Row():
@@ -453,7 +454,7 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
453
  outputs=[view3d, confidence_plot_image, cif_file]
454
  )
455
 
456
- @spaces.GPU
457
  def is_watermarked(file):
458
  #first initialize runner
459
  runner = InferenceRunner(configs)
 
430
  with gr.Row():
431
  view3d = Molecule3D(label="3D Visualization", reps=reps)
432
  legend = gr.Markdown("""
433
+ **Color Legend:**
434
+
435
+ - <span style="color:grey;">Gray: Unwatermarked Structure</span>
436
+ - <span style="color:cyan;">Cyan: Watermarked Structure</span>
437
+ """)
438
 
 
 
 
439
  with gr.Row():
440
  cif_file = gr.File(label="Download CIF File")
441
  with gr.Row():
 
454
  outputs=[view3d, confidence_plot_image, cif_file]
455
  )
456
 
457
+ @spaces.GPU(duration=120)
458
  def is_watermarked(file):
459
  #first initialize runner
460
  runner = InferenceRunner(configs)