Spaces:
Runtime error
Runtime error
File size: 687 Bytes
148aca7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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() |