set fine-tuned model

app.py

@@ -4,14 +4,20 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # Modell und Tokenizer laden
+HF_USER = "ai01firebird"
+MODEL_NAME = "emojinator-gpt2"
 
 # gpt2 outputs text!
 #tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
 #model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
 
 # distilgpt2 is only 80MB -> NOK, no emojis
-tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
-model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+#tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
+#model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+
+# fine-tuned
+model = AutoModelForCausalLM.from_pretrained(f"{HF_USER}/{MODEL_NAME}")
+tokenizer = AutoTokenizer.from_pretrained(f"{HF_USER}/{MODEL_NAME}")
 
 # tiny-gpt2 is only 20MB -> NOK, no emojis
 #tokenizer = AutoTokenizer.from_pretrained("sshleifer/tiny-gpt2")