Taaipoos

app.py

@@ -10,13 +10,6 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
 HF_TOKEN = os.environ['HF_TOKEN']
-DESCRIPTION = """# 🐐 GEITje-7B-chat 🐐
-## Een groot open Nederlands taalmodel
-
-[_Coming soon_](https://github.com/Rijgersberg/GEITje)"""
-
-if not torch.cuda.is_available():
-    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
 MAX_MAX_NEW_TOKENS = 2048
 DEFAULT_MAX_NEW_TOKENS = 1024
@@ -117,12 +110,12 @@ chat_interface = gr.ChatInterface(
         ["Kan je naar de maan fietsen?"],
         ["Wat is het belang van open source taalmodellen?"],
     ],
-    title="🐐 GEITje 7B Chat",
-    description="""Een eerste chatbot op basis van GEITje 7B: een groot open Nederlands taalmodel.
+    title="🐐 GEITje 7B Chat 🐐",
+    description="""Een eerste chatbot op basis van GEITje 7B: een groot open Nederlands taalmodel. Gemaakt voor demonstratiedoeleinden.
     
-    Dit is een chatbot gebaseerd op GEITje 7B, gemaakt voor demonstratiedoeleinden. Generatieve taalmodellen maken fouten, controleer daarom feiten voordat je ze overneemt. GEITJje Chat is niet uitgebreid getraind om _gealigned_ te zijn met menselijke waarden. Het is daarom mogelijk dat het problematische output genereert, zeker als het daartoe ge_prompt_ wordt.
+    Generatieve taalmodellen maken fouten, controleer daarom feiten voordat je ze overneemt. GEITje Chat is niet uitgebreid getraind om _gealigned_ te zijn met menselijke waarden. Het is daarom mogelijk dat het problematische output genereert, zeker als het daartoe _geprompt_ wordt.
     
-    Voor meer info over GEITJje: zie de <a href="https://github.com/Rijgersberg/GEITje">📄 README op GitHub</a>.""",
+    Voor meer info over GEITje: zie de <a href="https://github.com/Rijgersberg/GEITje">📄 README op GitHub</a>.""",
     submit_btn="Genereer",
     stop_btn="Stop",
     retry_btn="🔄 Opnieuw",