Update gen_table.py

gen_table.py

@@ -25,15 +25,11 @@ def model_size_flag(sz, FIELDS):
         return True
     if pd.isna(sz):
         return False
-    if '<4B' in FIELDS and sz < 4:
+    if '7B' in FIELDS and sz == 7:
         return True
-    if '4B-10B' in FIELDS and sz >= 4 and sz < 10:
+    if '13B' in FIELDS and sz == 13:
         return True
-    if '10B-20B' in FIELDS and sz >= 10 and sz < 20:
-        return True
-    if '20B-40B' in FIELDS and sz >= 20 and sz < 40:
-        return True
-    if '>40B' in FIELDS and sz >= 40:
+    if '70B' in FIELDS and sz == 70:
         return True
     return False