change to tiny-gpt2

app.py

@@ -10,10 +10,14 @@ 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")
 
+# conversion method
 def text_to_emoji(text):
     # remove special characters
     text_cleaned = re.sub(r"[.,!?;:]", "", text)