Update build_gradio_graph.py

build_gradio_graph.py

@@ -10,7 +10,6 @@ import shap
 import matplotlib.pyplot as plt
 from inference_polymers_gnn import predict
 import numpy as np
-import round
 
 # from datetime import datetime
 # import utils
@@ -58,7 +57,7 @@ def call_predict(inference_dict, cols_order, numerical_columns, target_columns):
         outputs = []
         for i in range(len(target_columns)):
             outputs += [y_pred[i][0]]
-            outputs += [round(np.random.uniform(2, 6), 1)]
+            outputs += [np.round(np.random.uniform(2, 6), 1)]
         # outputs += [fig]
 
         return outputs