Update app.py

app.py

@@ -1,9 +1,18 @@
-import streamlit as st
-from transformers import pipeline
+import gradio as gr
+import time
 
-pipe = pipeline('sentimate - analysis')
-text = st.text.area('enter some text')
+def echo(message, history, system_prompt, tokens):
+    response = f"System prompt: {system_prompt}\n Message: {message}."
+    for i in range(min(len(response), int(tokens))):
+        time.sleep(0.05)
+        yield response[: i+1]
 
-if text:
-    out = pipe(test)
-    st.json(out)
+demo = gr.ChatInterface(echo, 
+                        additional_inputs=[
+                            gr.Textbox("You are helpful AI.", label="System Prompt"), 
+                            gr.Slider(10, 100)
+                        ]
+                       )
+
+if __name__ == "__main__":
+    demo.queue().launch()