snikhilesh commited on
Commit
de9fb2d
·
verified ·
1 Parent(s): 41d2494

Upload backend/main.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 with current models
 
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.register_model_version(
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