Spaces:

Zaixi
/

FoldMark

Running on Zero

Zaixi commited on Feb 7

Commit

5efbcd3

1 Parent(s): 212c664

app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -450,8 +450,7 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
             outputs=[view3d, confidence_plot_image, cif_file]
         )
-    @torch.inference_mode()
-    @spaces.GPU(duration=60)
     def is_watermarked(file):
         # Generate a unique subdirectory and filename
         unique_id = str(uuid.uuid4())

             outputs=[view3d, confidence_plot_image, cif_file]
         )
+    @spaces.GPU
     def is_watermarked(file):
         # Generate a unique subdirectory and filename
         unique_id = str(uuid.uuid4())