File size: 620 Bytes
eaaba22 13441ce eaaba22 13441ce 7c63ba1 13441ce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import streamlit as st
from transformers import GPT2LMHeadModel, GPT2Tokenizer
#Load models
tokenizer = GPT2Tokenizer.from_pretrained("Legacy7070/Psychedelic-Trip-Report-Generator")
model = GPT2LMHeadModel.from_pretrained("Legacy7070/Psychedelic-Trip-Report-Generator")
#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)
|