Spaces:
Build error
Build error
import tensorflow as tf | |
import transformers | |
# Load the model | |
model = transformers.TFGPT2LMHeadModel.from_pretrained("souleater-diffusion.ckpt") | |
# Generate pictures using the model | |
def generate_picture(prompt): | |
input_ids = transformers.preprocessing.text.text_to_sequence(prompt, model.tokenizer.tokenize) | |
input_ids = tf.expand_dims(input_ids, 0) | |
output = model.generate(input_ids) | |
generated_text = model.tokenizer.decode(output[0], skip_special_tokens=True) | |
return generated_text | |
# GUI to enter the prompts | |
from tkinter import * | |
root = Tk() | |
root.title("souleater-diffusion.ckpt Model") | |
prompt_entry = Entry(root) | |
prompt_entry.pack() | |
def generate_callback(): | |
prompt = prompt_entry.get() | |
result = generate_picture(prompt) | |
result_label.config(text=result) | |
generate_button = Button(root, text="Generate", command=generate_callback) | |
generate_button.pack() | |
result_label = Label(root, text="") | |
result_label.pack() | |
root.mainloop() | |