change to distilgpt2

app.py

@@ -10,12 +10,12 @@ import torch
 #model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
 
 # distilgpt2 is only 80MB
-#tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
-#model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
+model = AutoModelForCausalLM.from_pretrained("distilgpt2")
 
-# tiny-gpt2 is only 20MB
-tokenizer = AutoTokenizer.from_pretrained("sshleifer/tiny-gpt2")
-model = AutoModelForCausalLM.from_pretrained("sshleifer/tiny-gpt2")
+# tiny-gpt2 is only 20MB -> NOK
+#tokenizer = AutoTokenizer.from_pretrained("sshleifer/tiny-gpt2")
+#model = AutoModelForCausalLM.from_pretrained("sshleifer/tiny-gpt2")
 
 # conversion method
 def text_to_emoji(input_text):