Update app.py

app.py

@@ -1,18 +1,14 @@
 
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import pipeline
 
-class EinfachPrompt:
-    def __init__(self):
-        self.tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-        self.model = GPT2LMHeadModel.from_pretrained("gpt2")
+# Erstelle ein prompt generator
+generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
 
-    def generate(self, prompt):
-        inputs = self.tokenizer.encode(prompt, return_tensors="pt")
-        outputs = self.model.generate(inputs, max_length=150, num_return_sequences=1, temperature=0.7)
-        generated = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
-        return generated
+# Frage den Benutzer nach einem Satzanfang
+prompt = input('Gib einen Satzanfang ein: ')
 
-if __name__ == "__main__":
-    einfach_prompt = EinfachPrompt()
-    prompt = "Erzähl mir etwas über EinfachPrompt."
-    print(einfach_prompt.generate(prompt))
+# Generiere einen neuen Text basierend auf dem Satzanfang
+text = generator(prompt, max_length=30, do_sample=True, temperature=0.7)[0]['generated_text']
+
+# Gib den generierten Text aus
+print(text)