Update app.py

app.py

@@ -1,19 +1,13 @@
-from transformers import AutoModelForSequenceClassification, AutoTokenizer, TextClassificationPipeline
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-MODEL_PATH = "results/checkpoint-6000/"  # Ändern Sie dies entsprechend
+MODEL_NAME = "gpt2"  # Ändern Sie dies entsprechend
 
-def load_model():
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
-    model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
-    pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
-    return pipeline
+tokenizer = GPT2Tokenizer.from_pretrained(MODEL_NAME)
+model = GPT2LMHeadModel.from_pretrained(MODEL_NAME)
 
-def classify_text(text):
-    pipeline = load_model()
-    result = pipeline(text)
-    return result
+prompt = "Was ist künstliche Intelligenz?"  # Ändern Sie dies entsprechend
+inputs = tokenizer.encode(prompt, return_tensors="pt")
+outputs = model.generate(inputs, max_length=200, num_return_sequences=5)
 
-if __name__ == "__main__":
-    text = input("Geben Sie einen Text ein: ")
-    result = classify_text(text)
-    print(result)
+for i, output in enumerate(outputs):
+    print(f"Output {i+1}: {tokenizer.decode(output)}")