Update app.py

app.py

@@ -9,6 +9,21 @@ from transformers import pipeline
 import ast
 
 
+################ CHAT BOT #################
+
+# Load the GPT model
+generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+
+# Streamlit chat UI
+st.title("GPT-3 Chatbox")
+
+user_input = st.text_input("You: ", "Hello, how are you?")
+
+if user_input:
+    response = generator(user_input, max_length=100, num_return_sequences=1)[0]['generated_text']
+    st.write(f"GPT-3: {response}")
+
+    
 ################ STATEMENT SUMMARIZATION #################
 
 # Load the summarization model