Update app.py

app.py

@@ -1,5 +1,6 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import streamlit as st
 
 # Load the GPT-2 large model and tokenizer
 model_name = "gpt2-large"
@@ -25,7 +26,14 @@ def generate_blog(topic_title, max_length=200):
     return blog_post
 
 # Example usage
-topic_title = input("Enter a topic title for the blog post: ")
-blog_post = generate_blog(topic_title)
-print("\nGenerated Blog Post:\n")
-print(blog_post)
+#topic_title = input("Enter a topic title for the blog post: ")
+#blog_post = generate_blog(topic_title)
+#print("\nGenerated Blog Post:\n")
+#print(blog_post)
+
+title = st.text_area('Enter title')
+if title:
+    out = generate_blog(title)
+    st.json(out)
+
+