change to distilgpt2

app.py

@@ -4,8 +4,15 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # Modell und Tokenizer laden
-tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
-model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
+
+# gpt2 outputs text!
+#tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
+#model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
+
+# distilgpt2 is only 80MB
+tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
+model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+
 
 def text_to_emoji(text):
     # remove special characters