Update app.py

app.py

@@ -115,8 +115,8 @@ 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/{jobname}/"))
-        return Molecule3D(f"{jobname}/boltz_results_{jobname}/predictions/{jobname}/{jobname}_model_0.pdb", label="Output", reps=representations)
+        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)
     except Exception as e:
         raise gr.Error(f"failed with error:{e}")