Update app.py

app.py

@@ -1,28 +1,9 @@
 import streamlit as st
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-# Title
-st.title("ChatGPT Clone")
+st.title("Chat UI")
 
-# Load pre-trained model and tokenizer
-model_name = 'gpt2'
-model = GPT2LMHeadModel.from_pretrained(model_name)
-tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+user_input = st.text_input("You: ")
 
-# Take user input
-input_text = st.text_input("You: ", "")
-
-# Encode input text and generate response
-if input_text:
-    # Encode the input text
-    input_ids = tokenizer.encode(input_text, return_tensors='pt')
-    # Generate a response
-    outputs = model.generate(input_ids, max_length=150, num_return_sequences=1, pad_token_id=tokenizer.eos_token_id)
-    # Decode the response
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    st.write("ChatGPT: ", response)
-
-# Run the Streamlit application
-if __name__ == '__main__':
-    st.sidebar.title("About")
-    st.sidebar.info("This is a simple ChatGPT clone using Hugging Face's Transformers library and Streamlit.")
+if st.button("Send"):
+    st.write("You: ", user_input)
+    st.write("Bot: Hello, How can I help you?")