|
from transformers import pipeline |
|
import textwrap |
|
import time |
|
import gradio as gr |
|
|
|
|
|
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() |
|
|