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

2 votes303 views1 answer
Python gui drag and drop
Python
0 vote304 views1 answer
Write a Python program to get the Fibonacci series between 0 to 50. Note : The Fibonacci Sequence is the series of numbers
Python
0 vote276 views1 answer
Iniciar servidor en Django
Python
1 vote321 views3 answers
Random choice sampling numpy
Python
0 vote354 views1 answer
Django insert bulk data
Python Django
0 vote317 views1 answer
Meter to cm in python
Python
0 vote302 views1 answer
How to round off values in columns in pandas in excel
Python
0 vote296 views1 answer
Calculer un temps en python
Python
0 vote322 views1 answer
If settings.debug
Python
0 vote332 views1 answer
Add x=y line to scatter plot python
Python