Update app.py
Browse files
app.py
CHANGED
|
@@ -1,9 +1,6 @@
|
|
| 1 |
import streamlit as st
|
| 2 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
| 3 |
|
| 4 |
-
# Set the title of the Streamlit app
|
| 5 |
-
st.title("GPT-2 Blog Post Generator")
|
| 6 |
-
|
| 7 |
# Initialize the tokenizer and model
|
| 8 |
model_name = 'gpt2-large'
|
| 9 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
|
@@ -36,7 +33,7 @@ if text:
|
|
| 36 |
except Exception as e:
|
| 37 |
st.error(f"An error occurred: {e}")
|
| 38 |
|
| 39 |
-
# Add instructions
|
| 40 |
-
st.write("""
|
| 41 |
-
|
| 42 |
-
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
| 3 |
|
|
|
|
|
|
|
|
|
|
| 4 |
# Initialize the tokenizer and model
|
| 5 |
model_name = 'gpt2-large'
|
| 6 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
|
|
|
| 33 |
except Exception as e:
|
| 34 |
st.error(f"An error occurred: {e}")
|
| 35 |
|
| 36 |
+
# # Add instructions
|
| 37 |
+
# st.write("""
|
| 38 |
+
# Enter a topic or a starting sentence in the text area above, and the GPT-2 model will generate a blog post for you.
|
| 39 |
+
# """)
|