Update app.py

app.py

@@ -1,6 +1,7 @@
-import gradio as gr
+import streamlit as st
 import requests
 
+# Function to query the Hugging Face model
 def query(data):
     headers = {
         "Content-Type": "application/json",
@@ -16,13 +17,19 @@ def query(data):
     result = response.json()
     return result.get("generated_text", "No response")
 
-iface = gr.Interface(
-    fn=query,
-    inputs=gr.inputs.Textbox(label="Input"),
-    outputs=gr.outputs.Textbox(label="Output"),
-    layout="vertical",
-    title="Hugging Face Chatbot",
-    description="Enter a message to chat with the model.",
-)
+# Streamlit app
+def main():
+    st.title("Hugging Face Chatbot")
+    st.write("Enter a message to chat with the model:")
 
-iface.launch(server_port=8889)
+    # Input text box
+    user_input = st.text_area("Your Message", "")
+
+    if st.button("Submit"):
+        if user_input:
+            st.write("User: ", user_input)
+            response = query(user_input)
+            st.write("Chatbot: ", response)
+
+if __name__ == "__main__":
+    main()