change to TinyLlama-1.1B-Chat-v1.0

app.py

@@ -9,14 +9,18 @@ import torch
 #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")
+# distilgpt2 is only 80MB -> NOK, no emojis
+#tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
+#model = AutoModelForCausalLM.from_pretrained("distilgpt2")
 
-# tiny-gpt2 is only 20MB -> NOK
+# tiny-gpt2 is only 20MB -> NOK, no emojis
 #tokenizer = AutoTokenizer.from_pretrained("sshleifer/tiny-gpt2")
 #model = AutoModelForCausalLM.from_pretrained("sshleifer/tiny-gpt2")
 
+# TinyLlama
+tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+
 # conversion method
 def text_to_emoji(input_text):
     # Eingabetext bereinigen (optional)