worker=0
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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)
|