Spaces:
Running
Running
Adding "Unknown" if num. params is None
#9
by
fzoll
- opened
app/backend/data_engine.py
CHANGED
@@ -104,6 +104,10 @@ class DataEngine:
|
|
104 |
models_list = self.models
|
105 |
|
106 |
df_model = pd.DataFrame(models_list)
|
|
|
|
|
|
|
|
|
107 |
|
108 |
# Create mapping for model names/aliases
|
109 |
if 'alias' in df_model.columns:
|
|
|
104 |
models_list = self.models
|
105 |
|
106 |
df_model = pd.DataFrame(models_list)
|
107 |
+
|
108 |
+
# Replace None values in num_params with "Unknown"
|
109 |
+
if 'num_params' in df_model.columns:
|
110 |
+
df_model['num_params'] = df_model['num_params'].fillna("Unknown")
|
111 |
|
112 |
# Create mapping for model names/aliases
|
113 |
if 'alias' in df_model.columns:
|