Update app.py

app.py

@@ -1,4 +1,16 @@
 import streamlit as st
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+#Load models
+repository_url = "https://huggingface.co/spaces/Legacy7070/Psychedelic-Trip-Report-Generator/tree/main/data"
+tokenizer = GPT2Tokenizer.from_pretrained(repository_url)
+model = GPT2LMHeadModel.from_pretrained(repository_url)
+
+#Set title and prompt
+st.title('Psychedelic Trip Report Generator')
+prompt = st.text_input("Enter a prompt for the trip report:")
+if prompt:
+    inputs = tokenizer.encode(prompt, return_tensors='pt')
+    outputs = model.generate(inputs, max_length=500, temperature=1.0)
+    generated_text = tokenizer.decode(outputs[0])
+    st.text(generated_text)