Spaces:
Running
Running
| from transformers import AutoModelForSequenceClassification, AutoTokenizer, TextClassificationPipeline | |
| MODEL_PATH = "results/checkpoint-6000/" # Ä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 | |
| def classify_text(text): | |
| pipeline = load_model() | |
| result = pipeline(text) | |
| return result | |
| if __name__ == "__main__": | |
| text = input("Geben Sie einen Text ein: ") | |
| result = classify_text(text) | |
| print(result) | |