Spaces:
Runtime error
Runtime error
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() |