diff --git a/data/gan.py b/data/gan.py
index 4a0fa48..ff51aa8 100644
--- a/data/gan.py
+++ b/data/gan.py
@@ -593,7 +593,7 @@ class Predict(GNet):
                                 #
                                 
                                 # df =   pd.DataFrame(np.round(f)).astype(np.int32)
-                                df =   pd.DataFrame(np.round(f),dtype=int)
+                                df =   pd.DataFrame(np.round(f),dtype=np.uint8)
                                 
                                 p = 0 not in df.sum(axis=1).values
                                 x = df.sum(axis=1).values