INNER CODE UNIT · Python
get_models
virajmavani/semi-auto-image-annotation-tool · web/backend/main.py:178
async def get_models():
"""Get list of available models"""
models_list = []
for model_id, model_info in AVAILABLE_MODELS.items():
models_list.append({
"id": model_id,
"name": model_info["name"],
"description": model_info["description"],
"framework": model_info["framework"],
"is_current": model_id == current_model_type,
"is_zero_shot": model_info.get("is_zero_shot", False),
})
return {
"models": models_list,
"current_model": current_model_type,
"current_threshold": current_threshold
}