Modif fonction

app.py

@@ -13,11 +13,8 @@ from modal_app import app, chai1_inference, download_inference_dependencies, her
 
 # Function to compute Chai1 inference
 def compute_Chai1(
-    force_redownload: bool = False,
     fasta_file: Optional[str] = None,
     inference_config_file: Optional[str] = None,
-    output_dir: Optional[str] = None,
-    run_id: Optional[str] = None,
 ):
     """Compute a Chai1 simulation.
 
@@ -29,6 +26,10 @@ def compute_Chai1(
     """
     with app.run():
         
+        force_redownload = False
+        output_dir = None
+        run_id = None
+        
         print("🧬 checking inference dependencies")
         download_inference_dependencies.remote(force=force_redownload)
 
@@ -54,7 +55,6 @@ def compute_Chai1(
 
         print(f"🧬 saving results to disk locally in {output_dir}")
         for ii, (scores, cif) in enumerate(results):
-            print(ii, scores)
             (Path(output_dir) / f"{run_id}-scores.model_idx_{ii}.npz").write_bytes(scores)
             (Path(output_dir) / f"{run_id}-preds.model_idx_{ii}.cif").write_text(cif)
         
@@ -68,11 +68,10 @@ def compute_Chai1(
 
 
 # Create the Gradio interface
-
 reps = [{"model": 0,"style": "cartoon","color": "whiteCarbon"}]
 
 with gr.Blocks() as demo:
-    inp = gr.Textbox(placeholder="Sequence file", label="Input Fasta file")
+    inp = gr.Textbox(placeholder="Fasta Sequence file", label="Input Fasta file")
     btn = gr.Button("Run")
     out = Molecule3D(label="Molecule3D", reps=reps)
     btn.click(fn=compute_Chai1, inputs=[inp], outputs=[out])

modal_app.py

@@ -1,6 +1,5 @@
-# Import necessary libraries
+# Import libraries
 from pathlib import Path
-
 import modal
 
 # Define parameters for the Modal app