Spaces:
Sleeping
Sleeping
File size: 303 Bytes
7858013 4305889 7858013 1c427fd 7858013 |
1 2 3 4 5 6 7 8 9 10 11 |
import gradio as gr
from transformers import pipeline
pipe = pipeline("text-generation")
def predict(prompt):
completion = pipe(prompt, max_length=50, num_return_sequences=1)[0]["generated_text"]
return completion
iface = gr.Interface(fn=predict, inputs="text", outputs="text")
iface.launch() |