Create app.py

app.py

@@ -0,0 +1,10 @@
+import streamlit as st
+from transformers import pipeline
+
+generator = pipeline(model="meta-llama/Llama-2-7b-chat-hf")
+prompt = st.text_area('Your prompt: ')
+res = generator(prompt)
+
+if res:
+    out = pipe(res)
+    st.json(out)