diff --git a/data/gan.py b/data/gan.py
index 382dc41..f2f3cdb 100644
--- a/data/gan.py
+++ b/data/gan.py
@@ -624,7 +624,7 @@ class Predict(GNet):
                         
                 
                 tf.compat.v1.reset_default_graph()
-                
+                df = pd.DataFrame(df)
                 return df.to_dict(orient='list')
                         # return df.to_dict(orient='list')
                         # count = str(len(os.listdir(self.out_dir)))