Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 from transformers import AutoTokenizer, pipeline, logging
 from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
 
@@ -36,4 +36,12 @@ pipe = pipeline(
     repetition_penalty=1.15
 )
 
-demo.launch()
+user_input = st.text_input("Input a phrase")
+
+prompt_template=f'''USER: {user_input}
+ASSISTANT:'''
+
+# Generate output when the "Generate" button is pressed
+if st.button("Generate the prompt"):
+    output = pipe(prompt_template)[0]['generated_text']
+    st.text_area("Prompt", value=output)