Generate python

from transformers import AutoTokenizer, AutoModelWithLMHead


tokenizer = AutoTokenizer.from_pretrained("Sentdex/GPyT")
model = AutoModelWithLMHead.from_pretrained("Sentdex/GPyT").to("cuda")


def generate(code, max_length=100):
    '''Takes input code, replaces newline chars with <N>, 
    tokenizes, feeds thru model, decodes, 
    then reformats the newlines back in'''
    newlinechar = "<N>"
    converted = code.replace("\n", newlinechar)
    tokenized = tokenizer.encode(converted, return_tensors='pt').to("cuda")
    resp = model.generate(tokenized, max_length=max_length).to("cuda")

    decoded = tokenizer.decode(resp[0])
    reformatted = decoded.replace("<N>","\n")
    return reformatted


print(generate("import"))

Source: pythonprogramming.net

Similar codes

7 votes95 views1 answer
Python get list of files in path
Python
4 votes118 views4 answers
Count unique elements in list python
Python
0 vote105 views2 answers
Cumulative chart python plotly
Python
0 vote123 views1 answer
Re_path(r'^logout/$',views.logout, {"next_page":'/'},name="logout"), AttributeError: module 'django.contrib.auth.views' has no attribute 'logout'
Python Django
0 vote97 views2 answers
Iqr in python
Python
0 vote103 views2 answers
To_csv create folder
Python
4 votes98 views3 answers
ArrayField in Django
Python
-1 vote91 views3 answers
Jama api python
Python
0 vote108 views1 answer
Subprocess ffmpeg x265 codec
Python
2 votes198 views1 answer
Sess.run tensorflow
Python