Spaces:
Runtime error
Runtime error
File size: 694 Bytes
41135d2 a644a56 1e263fe a644a56 9774338 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
import json
from transformers import AutoTokenizer, AutoModelWithLMHead
from transformers import pipeline
tokenizer = AutoTokenizer.from_pretrained("aditi2222/automatic_title_generation")
model = AutoModelWithLMHead.from_pretrained("aditi2222/automatic_title_generation")
generator = pipeline(
task="text-generation",
model=model,
tokenizer=tokenizer,
framework="tf"
)
model = AutoModelWithLMHead.from_pretrained("gpt2")
def green(prompt):
return generator(prompt, max_length=1000)
iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs=gr.outputs.Textbox(type="str", label="generated text"))
iface.launch() |