app

app.py

@@ -209,8 +209,8 @@ def create_protenix_json(input_data: Dict) -> List[Dict]:
     }]
 
 
-@torch.inference_mode()
-@spaces.GPU(duration=120)  # Specify a duration to avoid timeout
+#@torch.inference_mode()
+#@spaces.GPU(duration=120)  # Specify a duration to avoid timeout
 def predict_structure(input_collector: dict):
         """Handle both input types"""
         os.makedirs("./output", exist_ok=True)
@@ -450,7 +450,7 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
             outputs=[view3d, confidence_plot_image, cif_file]
         )
 
-    @spaces.GPU
+    #@spaces.GPU
     def is_watermarked(file):
         # Generate a unique subdirectory and filename
         unique_id = str(uuid.uuid4())