Create app.py

app.py

@@ -0,0 +1,16 @@
+import streamlit as st
+from gpt4all import GPT4All
+
+st.title("Write a story")
+
+model = GPT4All("mistral-7b-instruct-v0.1.Q4_0.gguf")
+
+prompt = st.text_area("Enter a prompt:", "Write a story about a guy with a mask")
+
+if st.button("Generate Text"):
+    output = model.generate(prompt)
+    sentences = output.split(".")[:-1]
+    formatted_output = ".\n".join(sentence.strip() for sentence in sentences if sentence.strip())
+    st.write("Generated Story:")
+    st.text(formatted_output + ".")
+    st.session_state['story'] = formatted_output