From 1e5e56fa2eef8f1b191ba6938685537e06c434e4 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 21 Jan 2023 22:13:01 -0300
Subject: [PATCH] Better recognize the 4chan model (for #19)

---
 server.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server.py b/server.py
index bff78828..67320377 100644
--- a/server.py
+++ b/server.py
@@ -155,7 +155,7 @@ def formatted_outputs(reply, model_name):
         if model_name.lower().startswith('galactica'):
             reply = fix_galactica(reply)
             return reply, reply, generate_basic_html(reply)
-        elif model_name.lower().startswith('gpt4chan'):
+        elif model_name.lower().startswith(('gpt4chan', 'gpt-4chan', '4chan')):
             reply = fix_gpt4chan(reply)
             return reply, 'Only applicable for GALACTICA models.', generate_4chan_html(reply)
         else: