Update app.py

app.py

@@ -3,14 +3,16 @@ import os
 import spaces
 import gradio as gr 
 
-from allmetal3d.utils.helpersNew import *
-from allmetal3d.utils.main import predict, predict_cli, update_mode
+
 
 from gradio_molecule3d import Molecule3D
 
 os.system("pip install allmetal3d-0.4.0-py3-none-any.whl")
 
 
+from allmetal3d.utils.helpersNew import *
+from allmetal3d.utils.main import predict, predict_cli, update_mode
+
 @spaces.GPU
 def predict_zero_gpu(pdb, models, pthreshold, threshold, batch_size, mode, central_residue, radius):
     out, metal_pdb, metal_cube, water_pdb, water_cube, results_json, waters_json = predict(pdb, models, pthreshold, threshold, batch_size, mode, central_residue, radius)