speech_app / main1.py
krishna34-aimlcurious's picture
Upload 2 files
148aca7 verified
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()