Chatoos / app.py
WesanCZE's picture
Create app.py
4f7ca9c verified
raw
history blame contribute delete
841 Bytes
import gradio as gr
from transformers import pipeline
import pyttsx3
# Inicializace modelu Hugging Face pro generování textu
generator = pipeline("text-generation", model="gpt2")
# TTS: Funkce pro syntézu hlasu
engine = pyttsx3.init()
def speak(text):
engine.say(text)
engine.runAndWait()
# Funkce pro zpracování uživatelského vstupu
def chatbot(user_input):
response = generator(user_input, max_length=50, num_return_sequences=1)
generated_text = response[0]["generated_text"]
speak(generated_text) # Přečte odpověď
return generated_text
# Vytvoření jednoduchého uživatelského rozhraní
interface = gr.Interface(
fn=chatbot,
inputs=gr.Textbox(label="Zadej svou otázku:"),
outputs=gr.Textbox(label="Odpověď od chatbota:"),
live=True
)
# Spuštění aplikace
interface.launch()