CoolFace
Apppublic

AGayathriiii/DotNetJavaCodeConversion

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes
model_router.py14 linesDownload Raw Back to backend
1from backend.mistral7b import convert_with_mistral_7b2from backend.mistrallarge2 import convert_with_mistral_large23from backend.codegen2 import convert_with_codegen24 5def route_model_conversion(model_name, cs_code):6    if "Mistral-7B" in model_name:7        return convert_with_mistral_7b(cs_code)8    elif "Mixtral" in model_name:9        return convert_with_mistral_large2(cs_code)10    elif "codegen2" in model_name:11        return convert_with_codegen2(cs_code)12    else:13        return "Unsupported model selected."14