File size: 563 Bytes
a98fecc 3ad1bbb dae99b3 a98fecc 0b5acf9 dae99b3 a98fecc 0d8fbad da37131 dae99b3 a98fecc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from transformers import pipeline
import textwrap
import time
import gradio as gr
# Load a pre-trained chatbot model
chatbot = pipeline('text-generation', model='gpt2-large')
def wrap(x):
return textwrap.fill(x, replace_whitespace=False, fix_sentence_endings=True)
def chat(input_text, additional_param):
generated_text = chatbot(input_text)[0]["generated_text"]
print(f"Generated text: {generated_text}")
print(f"Additional parameter: {additional_param}")
return generated_text
gr.Interface(fn=chat, inputs="text", outputs="text").launch()
|