omair14 commited on
Commit
80d20fa
·
verified ·
1 Parent(s): c38ee55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,6 +4,9 @@ from transformers import pipeline
4
  # Load the text generation pipeline with gpt2-large
5
  try:
6
  generator = pipeline('text-generation', model='gpt2-large')
 
 
 
7
 
8
  def generate_blog_post(topic):
9
  try:
@@ -20,4 +23,4 @@ topic = st.text_input("Enter the topic for the blog post:")
20
  if st.button("Generate"):
21
  with st.spinner('Generating blog post...'):
22
  blog_post = generate_blog_post(topic)
23
- st.write(blog_post)
 
4
  # Load the text generation pipeline with gpt2-large
5
  try:
6
  generator = pipeline('text-generation', model='gpt2-large')
7
+ st.write("Model loaded successfully.")
8
+ except Exception as e:
9
+ st.write(f"Error loading model: {e}")
10
 
11
  def generate_blog_post(topic):
12
  try:
 
23
  if st.button("Generate"):
24
  with st.spinner('Generating blog post...'):
25
  blog_post = generate_blog_post(topic)
26
+ st.write(blog_post)