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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,13 +1,13 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- @st.cache_resource(allow_output_mutation=True)
5
  def load_model():
6
  return pipeline("text-generation", model="PeterBrendan/pbjs_gpt2")
7
 
8
  def main():
9
- st.title("Hugging Face Model Demo")
10
- st.write("Enter some text and get generated text as output.")
11
 
12
  # Create a text input field for user input
13
  user_input = st.text_input("Enter text:", "")
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ @st.cache_resource
5
  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:", "")