update environment and boltz2 version

app.py

@@ -116,12 +116,12 @@ def predict(jobname, inputs, recycling_steps, sampling_steps, diffusion_samples)
         os.system(f"boltz predict {jobname}/{jobname}.yaml --use_msa_server --out_dir {jobname} --recycling_steps {recycling_steps} --sampling_steps {sampling_steps} --diffusion_samples {diffusion_samples} --override --output_format pdb")
         print(os.listdir(jobname))
         print(os.listdir(f"{jobname}/boltz_results_{jobname}/predictions/"))
-        return Molecule3D(f"{jobname}/boltz_results_{jobname}/predictions/{jobname}_model_0.cif", label="Output", reps=representations)
+        return Molecule3D(f"{jobname}/boltz_results_{jobname}/predictions/{jobname}/{jobname}_model_0.cif", label="Output", reps=representations)
     except Exception as e:
         raise gr.Error(f"failed with error:{e}")
 
 with gr.Blocks() as blocks:
-    gr.Markdown("# Boltz-1")
+    gr.Markdown("# Boltz")
     gr.Markdown("""Open GUI for running [Boltz-1 model](https://github.com/jwohlwend/boltz/) <br>
     Key components:
     - MMSeqs2 Webserver [Mirdita et al.](https://www.nature.com/articles/s41592-022-01488-1)

requirements.txt

@@ -1,3 +1,3 @@
-boltz
+boltz[cuda]
 gradio_molecule3d
 gradio_cofoldinginput