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

24 votes272 views1 answer
How to replace na values in python
Python
2 votes269 views2 answers
Python get time difference in milliseconds
Python
0 vote279 views1 answer
ApiError('d984ac0ae28d4afda0ddb26e348db535', UploadError('path', UploadWriteFailed(reason=WriteError('malformed_path', None), upload_session_id='AAAAAAAAACLukDh6ck8ZUw'))) python
Python
12 votes268 views1 answer
Create new env in anaconda
Python
19 votes275 views5 answers
Python floor
Python
14 votes269 views2 answers
How to convert csv to excel in python
Python
0 vote245 views1 answer
In <lambda> tk.Button(self.frame, text = text, command= lambda:self.new_window(number, _class)).grid(padx= 5, pady= 5, sticky= N + E + W + S) File "C:\Users\CSB\Desktop\python program\triple_window.py", line 50, in new_window
Python
1 vote370 views3 answers
Ipynb to py online converter
Python
10 votes255 views1 answer
Add static file in django
Python
24 votes277 views2 answers
How to check the type of a variable in python
Python