Code_Summarization / models.py
nam194's picture
Create models.py
65e33b6
raw
history blame
179 Bytes
def return_models(language):
langs = ["php", "python", "java", "ruby", "javascript"]
if language in langs:
return f"nam194/cotet5sum_{language}"
return ""