File size: 251 Bytes
700c096
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from gpt2_client import *

gpt2 = GPT2()

streamlit_code_base = gpt2.generate(
    prompt="Enter prompt here",
    temperature=0.7,
    top_p=0.9,
    nsamples=1,
    batch_size=1,
    length=1000,
    include_prefix=True
)

print(streamlit_code_base)