|
|
import gradio as gr |
|
|
from transformers import pipeline |
|
|
import pyttsx3 |
|
|
|
|
|
|
|
|
generator = pipeline("text-generation", model="gpt2") |
|
|
|
|
|
|
|
|
engine = pyttsx3.init() |
|
|
def speak(text): |
|
|
engine.say(text) |
|
|
engine.runAndWait() |
|
|
|
|
|
|
|
|
def chatbot(user_input): |
|
|
response = generator(user_input, max_length=50, num_return_sequences=1) |
|
|
generated_text = response[0]["generated_text"] |
|
|
speak(generated_text) |
|
|
return generated_text |
|
|
|
|
|
|
|
|
interface = gr.Interface( |
|
|
fn=chatbot, |
|
|
inputs=gr.Textbox(label="Zadej svou otázku:"), |
|
|
outputs=gr.Textbox(label="Odpověď od chatbota:"), |
|
|
live=True |
|
|
) |
|
|
|
|
|
|
|
|
interface.launch() |