Update app.py

app.py

@@ -1,3 +1,4 @@
+import streamlit as st 
 import transformers
 import torch 
 from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
@@ -23,7 +24,7 @@ def generate_response(user_input):
 # Start the chat loop
 while True:
   # Get user input
-  uinput = input("Enter your input: ")
+  uinput = st.text_input("Enter your input: ")
 
   if uinput.lower() == "quit": 
     break