speech_app / app.py
krishna34-aimlcurious's picture
Update app.py
de57b5d verified
raw
history blame contribute delete
669 Bytes
from kivy.app import App
from kivy.uix.boxlayout import BoxLayout
from transformers import pipeline
# Load the GPT-2 model
generator = pipeline("text-generation", model="gpt2")
class TextGenLayout(BoxLayout):
def generate(self):
user_prompt = self.ids.prompt_input.text
if not user_prompt.strip():
self.ids.output_box.text = "⚠️ Please enter some text."
return
result = generator(user_prompt, max_length=80, num_return_sequences=1)
generated_text = result[0]["generated_text"]
self.ids.output_box.text = generated_text
class TextGenApp(App):
def build(self):
return TextGenLayout()