Update app.py

app.py

@@ -1,3 +1,7 @@
+import streamlit as st
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import torch
+
 # Load the GPT-2 model and tokenizer
 model_name = 'gpt2-large'
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
@@ -23,4 +27,4 @@ topic = st.text_input("Topic:")
 
 if st.button("Generate"):
     blog_post = generate_blog_post(topic)
-    st.write(blog_post)
+    st.write(blog_post)