Update gen_table.py

gen_table.py

@@ -32,14 +32,16 @@ def model_size_flag(sz, FIELDS):
     if '70B' in FIELDS and sz == 70:
         return True
     return False
-
+    
 
 def model_type_flag(line, FIELDS):
-    if 'OpenSource' in FIELDS and line['OpenSource'] == 'Yes':
+    if 'Commercial LLMs' in FIELDS and line['Commercial LLMs'] == 'Yes':
+        return True
+    if 'General LLMs' in FIELDS and line['General LLMs'] == 'Yes':
         return True
-    if 'API' in FIELDS and line['OpenSource'] == 'No' and line['Verified'] == 'Yes':
+    if 'Medical LLMs' in FIELDS and line['Medical LLMs'] == 'Yes':
         return True
-    if 'Proprietary' in FIELDS and line['OpenSource'] == 'No' and line['Verified'] == 'No':
+    if 'SOTA' in FIELDS and line['SOTA'] == 'Yes':
         return True
     return False