Update app.py

app.py

@@ -19,7 +19,7 @@ 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)
+    return predict(pdb, models, pthreshold, threshold, batch_size, mode, central_residue, radius)
 
 with gr.Blocks() as blocks: 
     gr.Markdown("## AllMetal3D and Water3D")