File size: 386 Bytes
f6163ea f992af0 e08bfbf f992af0 546ccd4 93303ed 546ccd4 e08bfbf 546ccd4 e08bfbf be94d83 ffb04d2 93303ed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import torch
from transformers import pipeline
import gradio as gr
def generating_text(txt):
device = torch.device("cpu")
pipe = pipeline(
"text-generation", model="checkpoint-5000", device=device
)
return pipe(txt, num_return_sequences=1)[0]["generated_text"]
demo = gr.Interface(fn=generating_text, inputs="text", outputs="text")
demo.launch(inline=False)
|