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)