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

-3 votes334 views2 answers
Text detection from image using opencv python
Python
0 vote334 views1 answer
Add favicon in django admin
Python
0 vote337 views3 answers
Pyqt message box set information text
Python
0 vote397 views1 answer
Python string: .split()
Python
31 votes359 views6 answers
Python dataframe sort by column name
Python
4 votes321 views1 answer
Pandas fill empty
Python
4 votes338 views3 answers
Gui python
Python
12 votes332 views3 answers
Rstrip python
Python
3 votes358 views1 answer
Display flask across network
Python Flask
7 votes362 views2 answers
Add image to jupyter notebook in markdown
Python