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

0 vote364 views1 answer
How do you say not equal to python
Python
32 votes304 views3 answers
Python move files
Python
20 votes315 views4 answers
How to make computer go in sleep mode using pythn
Python
1 vote327 views1 answer
Discord.py say something
Python
1 vote300 views1 answer
Python imaginary numbers
Python
0 vote313 views3 answers
Python with quick sort
Python
1 vote316 views1 answer
Python print dict new line
Python
0 vote322 views1 answer
Unsupervised knn
Python
0 vote331 views1 answer
Python how to turn a word into a list
Python
3 votes310 views1 answer
N random numbers python
Python