PeterBrendan commited on
Commit
177ebcb
·
1 Parent(s): e474d15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,8 +6,8 @@ def load_model():
6
  return pipeline("text-generation", model="PeterBrendan/pbjs_gpt2")
7
 
8
  def main():
9
- st.title("Pbjs GPT")
10
- st.write("Enter some text like **bidderTimoeut** and get a generated Prebid config output.")
11
 
12
  # Create a text input field for user input
13
  user_input = st.text_input("Enter text:", "")
@@ -18,7 +18,7 @@ def main():
18
  generator = load_model()
19
 
20
  # Generate text based on user input
21
- generated_text = generator(user_input, max_length=100, num_return_sequences=1)[0]["generated_text"]
22
 
23
  # Display the generated text
24
  st.write("Generated Text:")
 
6
  return pipeline("text-generation", model="PeterBrendan/pbjs_gpt2")
7
 
8
  def main():
9
+ st.title("Pbjs GPT2")
10
+ st.write("Enter some text like **bidderTimeout** and get a generated Prebid config output.")
11
 
12
  # Create a text input field for user input
13
  user_input = st.text_input("Enter text:", "")
 
18
  generator = load_model()
19
 
20
  # Generate text based on user input
21
+ generated_text = generator(user_input, max_length=500, num_return_sequences=1)[0]["generated_text"]
22
 
23
  # Display the generated text
24
  st.write("Generated Text:")