Description and back to fully trained geitje-chat

app.py

@@ -10,7 +10,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
 HF_TOKEN = os.environ['HF_TOKEN']
-DESCRIPTION = """# 🐐 GEITje 7B Chat 🐐
+DESCRIPTION = """# 🐐 GEITje-7B-chat 🐐
 ## Een groot open Nederlands taalmodel
 
 [_Coming soon_](https://github.com/Rijgersberg/GEITje)"""
@@ -23,7 +23,7 @@ DEFAULT_MAX_NEW_TOKENS = 1024
 MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
 
 if torch.cuda.is_available():
-    model_id = "Rijgersberg/GEITje-7B-chat-epoch2"
+    model_id = "Rijgersberg/GEITje-7B-chat"
     model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto", token=HF_TOKEN)
     tokenizer = AutoTokenizer.from_pretrained(model_id, token=HF_TOKEN)