Upload backend/main.py with huggingface_hub
Browse files- backend/main.py +6 -2
backend/main.py
CHANGED
|
@@ -192,7 +192,8 @@ async def startup_event():
|
|
| 192 |
"alert_threshold": 0.05
|
| 193 |
})
|
| 194 |
|
| 195 |
-
# Initialize versioning system
|
|
|
|
| 196 |
model_versions = [
|
| 197 |
{"model_id": "bio_clinical_bert", "version": "1.0.0", "source": "HuggingFace"},
|
| 198 |
{"model_id": "biogpt", "version": "1.0.0", "source": "HuggingFace"},
|
|
@@ -202,10 +203,13 @@ async def startup_event():
|
|
| 202 |
{"model_id": "medgemma_2b", "version": "1.0.0", "source": "HuggingFace"}
|
| 203 |
]
|
| 204 |
|
|
|
|
| 205 |
for model_config in model_versions:
|
| 206 |
-
versioning_system.
|
| 207 |
model_id=model_config["model_id"],
|
| 208 |
version=model_config["version"],
|
|
|
|
|
|
|
| 209 |
metadata={"source": model_config["source"]}
|
| 210 |
)
|
| 211 |
|
|
|
|
| 192 |
"alert_threshold": 0.05
|
| 193 |
})
|
| 194 |
|
| 195 |
+
# Initialize versioning system - default models already loaded in constructor
|
| 196 |
+
# Additional models can be registered here if needed
|
| 197 |
model_versions = [
|
| 198 |
{"model_id": "bio_clinical_bert", "version": "1.0.0", "source": "HuggingFace"},
|
| 199 |
{"model_id": "biogpt", "version": "1.0.0", "source": "HuggingFace"},
|
|
|
|
| 203 |
{"model_id": "medgemma_2b", "version": "1.0.0", "source": "HuggingFace"}
|
| 204 |
]
|
| 205 |
|
| 206 |
+
# Register additional models using the correct method
|
| 207 |
for model_config in model_versions:
|
| 208 |
+
versioning_system.model_registry.register_model(
|
| 209 |
model_id=model_config["model_id"],
|
| 210 |
version=model_config["version"],
|
| 211 |
+
model_name=model_config["model_id"].replace("_", " ").title(),
|
| 212 |
+
model_path=f"huggingface/{model_config['model_id']}",
|
| 213 |
metadata={"source": model_config["source"]}
|
| 214 |
)
|
| 215 |
|