Spaces:
Running
Running
Update src/aibom_generator/generator.py
Browse files
src/aibom_generator/generator.py
CHANGED
@@ -120,7 +120,7 @@ class AIBOMGenerator:
|
|
120 |
metadata["ai:framework"] = "PyTorch" if "transformers" in metadata.get("library_name", "") else "Unknown"
|
121 |
|
122 |
|
123 |
-
|
124 |
|
125 |
def _extract_unstructured_metadata(self, model_card: ModelCard) -> Dict[str, Any]:
|
126 |
return {}
|
|
|
120 |
metadata["ai:framework"] = "PyTorch" if "transformers" in metadata.get("library_name", "") else "Unknown"
|
121 |
|
122 |
|
123 |
+
return {k: v for k, v in metadata.items() if v is not None}
|
124 |
|
125 |
def _extract_unstructured_metadata(self, model_card: ModelCard) -> Dict[str, Any]:
|
126 |
return {}
|