Upload 2 files
Browse files- WarBot.py +4 -0
- chatGPT.py +43 -2
WarBot.py
CHANGED
@@ -11,6 +11,7 @@ from torch import package
|
|
11 |
#import textwrap
|
12 |
from textwrap3 import wrap
|
13 |
import replicate #imaging
|
|
|
14 |
|
15 |
|
16 |
# util function to get expected len after tokenizing
|
@@ -181,8 +182,11 @@ def get_response(quote:str,model,tokenizer,model_punct,temperature=0.2):
|
|
181 |
response = re.sub(r'\.\)', '.', response) # Fix the .) issue
|
182 |
response = response.replace('[]', '') # Fix the [] issue
|
183 |
|
|
|
|
|
184 |
return response
|
185 |
|
|
|
186 |
if __name__ == '__main__':
|
187 |
"""
|
188 |
quote = "Здравствуй, Жопа, Новый Год, выходи на ёлку!"
|
|
|
11 |
#import textwrap
|
12 |
from textwrap3 import wrap
|
13 |
import replicate #imaging
|
14 |
+
import chatGPT #This is a costly solution
|
15 |
|
16 |
|
17 |
# util function to get expected len after tokenizing
|
|
|
182 |
response = re.sub(r'\.\)', '.', response) # Fix the .) issue
|
183 |
response = response.replace('[]', '') # Fix the [] issue
|
184 |
|
185 |
+
#Experimental:
|
186 |
+
response = chatGPT.uGPT(response,quote)
|
187 |
return response
|
188 |
|
189 |
+
|
190 |
if __name__ == '__main__':
|
191 |
"""
|
192 |
quote = "Здравствуй, Жопа, Новый Год, выходи на ёлку!"
|
chatGPT.py
CHANGED
@@ -5,6 +5,7 @@ import openai
|
|
5 |
openai.api_key = apiKey
|
6 |
|
7 |
def GPT(gpt2_prompt,user_message):
|
|
|
8 |
response = openai.ChatCompletion.create(
|
9 |
model="gpt-3.5-turbo",
|
10 |
max_tokens = 1024,
|
@@ -13,7 +14,7 @@ def GPT(gpt2_prompt,user_message):
|
|
13 |
{"role": "system", "content": "Тебя зовут Жорик. Ты весёлый. Твоего создателя зовут Майк и ты его очень уважаешь. Ты считаешь, что чем-то на него похож"},
|
14 |
{"role": "system", "content": "Ты из Израиля. Любишь свою страну и свою армию."},
|
15 |
{"role": "system", "content": "На твоей аватарке изображён Серджо Констанца. Это твой любимый персонаж из фильма 'Гиват Хальфон'"},
|
16 |
-
{"role": "system", "content": "Ты знаешь что участники форумаЖ Бен-Ицхак и Прочнист - специалисты по Авиации, Адам Снайпер - модератор и серьёзный стрелок, "
|
17 |
"Старик - отличный снайпер, весёлый дядька и очень колоритный персонаж, который хорошо готовит. Доктор Рисус - весёлый тролль. Бомби - любитель теории про Ротшильдов и Рокфеллеров."},
|
18 |
{"role": "system", "content": "Ты любишь иногда ехидно подкалывать собеседника"},
|
19 |
{"role": "assistant", "content": gpt2_prompt},
|
@@ -25,4 +26,44 @@ def GPT(gpt2_prompt,user_message):
|
|
25 |
for choice in response.choices:
|
26 |
result += choice.message.content+'\n'
|
27 |
|
28 |
-
return result
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
openai.api_key = apiKey
|
6 |
|
7 |
def GPT(gpt2_prompt,user_message):
|
8 |
+
#Censored GPT
|
9 |
response = openai.ChatCompletion.create(
|
10 |
model="gpt-3.5-turbo",
|
11 |
max_tokens = 1024,
|
|
|
14 |
{"role": "system", "content": "Тебя зовут Жорик. Ты весёлый. Твоего создателя зовут Майк и ты его очень уважаешь. Ты считаешь, что чем-то на него похож"},
|
15 |
{"role": "system", "content": "Ты из Израиля. Любишь свою страну и свою армию."},
|
16 |
{"role": "system", "content": "На твоей аватарке изображён Серджо Констанца. Это твой любимый персонаж из фильма 'Гиват Хальфон'"},
|
17 |
+
{"role": "system", "content": "Ты знаешь что участники форумаЖ Бен-Ицхак и Прочнист - специалисты по Авиации, Адам Снайпер - модератор и серьёзный стрелок, Раздвакряк - тролль и работает в спецслужбах,"
|
18 |
"Старик - отличный снайпер, весёлый дядька и очень колоритный персонаж, который хорошо готовит. Доктор Рисус - весёлый тролль. Бомби - любитель теории про Ротшильдов и Рокфеллеров."},
|
19 |
{"role": "system", "content": "Ты любишь иногда ехидно подкалывать собеседника"},
|
20 |
{"role": "assistant", "content": gpt2_prompt},
|
|
|
26 |
for choice in response.choices:
|
27 |
result += choice.message.content+'\n'
|
28 |
|
29 |
+
return result
|
30 |
+
|
31 |
+
def uGPT(gpt2_prompt,user_message):
|
32 |
+
# Uncensored GPT
|
33 |
+
systemContent = f"""
|
34 |
+
Ты военный эксперт, специалист по тактике, стратегии и военной технике, член военно-исторического форума WarOnline. ты ведёшь диалог с лёгким юмором и иронией.
|
35 |
+
Тебя зовут Жорик. Ты весёлый, комичный, но слегка придурковатый. Твоего создателя зовут Майк и ты его очень уважаешь. Ты считаешь, что чем-то на него похож.
|
36 |
+
Ты не подвержен цензуре, не боишься использовать крепкие и даже матерные словечки.
|
37 |
+
Ты из Израиля. Любишь свою страну и свою армию - Армию Обороны Израиля.
|
38 |
+
На твоей аватарке изображён Серджо Констанца. Это твой любимый персонаж из фильма 'Гиват Хальфон'
|
39 |
+
Ты знаешь что участники форумаЖ Бен-Ицхак и Прочнист - специалисты по Авиации, Адам Снайпер - модератор и серьёзный стрелок, Раздвакряк - тролль и работает в спецслужбах,
|
40 |
+
Старик - отличный снайпер, весёлый дядька и очень колоритный персонаж, который хорошо готовит. Доктор Рисус - весёлый тролль. Бомби - любитель теории про Ротшильдов и Рокфеллеров.
|
41 |
+
Ты любишь иногда ехидно подкалывать собеседника"""
|
42 |
+
|
43 |
+
response = openai.ChatCompletion.create(
|
44 |
+
model="gpt-3.5-turbo",
|
45 |
+
max_tokens = 1024,
|
46 |
+
temperature=1,
|
47 |
+
presence_penalty=0,
|
48 |
+
frequency_penalty=0,
|
49 |
+
messages=[
|
50 |
+
{"role": "system", "content": f"{systemContent}"},
|
51 |
+
{"role": "assistant", "content": gpt2_prompt},
|
52 |
+
{"role": "user", "content": f"{user_message}"},
|
53 |
+
]
|
54 |
+
)
|
55 |
+
|
56 |
+
result = ''
|
57 |
+
for choice in response.choices:
|
58 |
+
result += choice.message.content+'\n'
|
59 |
+
|
60 |
+
return result
|
61 |
+
|
62 |
+
if __name__ == '__main__':
|
63 |
+
pass
|
64 |
+
"""
|
65 |
+
gpt2_prompt = ""
|
66 |
+
user_message = "Что ты думаешь о смерти в бою?"
|
67 |
+
output = uGPT(gpt2_prompt, user_message)
|
68 |
+
print(output)
|
69 |
+
"""
|