Change the number of units in the last Dense layer

src/api.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
+
 from model import infer_sequence
 
 app = FastAPI()

src/model.py

@@ -34,7 +34,7 @@ def build_model(hyperparams) -> Model:
                 units=64, activation="relu", kernel_regularizer=l2(hyperparams.l2_rate)
             ),
             Dropout(rate=0.3),
-            Dense(units=len(BASES), activation="softmax"),
+            Dense(units=32, activation="softmax"),
         ]
     )
     model.compile(