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