worker=0

app.py

@@ -430,11 +430,12 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
         with gr.Row():
             view3d = Molecule3D(label="3D Visualization", reps=reps)
         legend = gr.Markdown("""
-        **Color Legend:**
+            **Color Legend:**
+
+            - <span style="color:grey;">Gray: Unwatermarked Structure</span>
+            - <span style="color:cyan;">Cyan: Watermarked Structure</span>
+            """)
 
-        - <span style="color:grey">Unwatermarked Structure</span>
-        - <span style="color:cyan">Watermarked Structure</span>
-        """)
         with gr.Row():
             cif_file = gr.File(label="Download CIF File")
         with gr.Row():
@@ -453,7 +454,7 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
             outputs=[view3d, confidence_plot_image, cif_file]
         )
 
-    @spaces.GPU
+    @spaces.GPU(duration=120)
     def is_watermarked(file):
         #first initialize runner
         runner = InferenceRunner(configs)