Update app.py

app.py

@@ -281,7 +281,7 @@ def flex_seq(input_seq, input_file):
         print("Falling back to CPU execution. This may take a while...")
         return core_flex_seq(input_seq, input_file, force_cpu=True)
 
-def core_flex_3d(input_file):        
+def core_flex_3d(input_file, force_cpu=False):        
     if not input_file:
         return None, "Provide a file or a input sequence"
     
@@ -467,23 +467,23 @@ def core_flex_3d(input_file):
     return output_files, output_message, output_files_enm
 
 @spaces.GPU
-def flex_3d_gpu(input_seq, input_file):
+def flex_3d_gpu(input_file):
     """Wrapper that requests ZeroGPU."""
-    return core_flex_3d(input_seq, input_file, force_cpu=False)
+    return core_flex_3d(input_file, force_cpu=False)
 
-def flex_3d(input_seq, input_file):
+def flex_3d(input_file):
     """
     Main entry point for Gradio. 
     Tries GPU first, falls back to CPU if quota is exceeded or time limit runs out.
     """
     try:
         print("Attempting to run on ZeroGPU...")
-        return flex_3d_gpu(input_seq, input_file)
+        return flex_3d_gpu(input_file)
     except Exception as e:
         # ZeroGPU exceptions (like SpaceTaskError or timeouts) are caught here
         print(f"ZeroGPU failed or timed out. Reason: {e}")
         print("Falling back to CPU execution. This may take longer...")
-        return core_flex_3d(input_seq, input_file, force_cpu=True)
+        return core_flex_3d(input_file, force_cpu=True)
 
     
 def rescale_bfactors(pdb_file):