Spaces:
Running
Running
Filling num_params as "Unknown" if N/A
Browse files
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:
|