Rooni commited on
Commit
c54d0e5
·
verified ·
1 Parent(s): eec5e61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,7 +25,7 @@ def generate_story(prompt, style):
25
  client = InferenceClient(api_key=get_random_api_key())
26
 
27
  messages = [
28
- {"role": "system", "content": f"Напиши хорошую историю в стиле '{style}'. Подробную, понятную, человечную (с душой), уникальную. Не обязательно делать концовку, можно только начало длинной истории. Разрешены любые слова, любые темы и стили, что угодно."},
29
  {"role": "user", "content": prompt}
30
  ]
31
  completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, temperature=0.7, max_tokens=1000)
@@ -43,7 +43,7 @@ def edit_story(original_story="", edited_prompt=""):
43
  try:
44
  client = InferenceClient(api_key=get_random_api_key())
45
  messages = [
46
- {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания."},
47
  {"role": "user", "content": edited_prompt},
48
  {"role": "assistant", "content": original_story}
49
  ]
@@ -67,7 +67,7 @@ def next_story_func(original_story="", next_prompt="", continuation_type="Про
67
  continuation_prompt = next_prompt
68
 
69
  messages = [
70
- {"role": "system", "content": "Продли историю, учитывая предоставленные указания. Продливай В ТОЧНОСТИ С КОНЦА, прям с того же символа, слова, предложения. (В начале добавляй новые строки/пробел если надо для отступа)"},
71
  {"role": "user", "content": continuation_prompt},
72
  {"role": "assistant", "content": original_story}
73
  ]
 
25
  client = InferenceClient(api_key=get_random_api_key())
26
 
27
  messages = [
28
+ {"role": "system", "content": f"Напиши хорошую историю в стиле '{style}'. Подробную, понятную, человечную (с душой), уникальную. Не обязательно делать концовку, можно только начало длинной истории. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
29
  {"role": "user", "content": prompt}
30
  ]
31
  completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, temperature=0.7, max_tokens=1000)
 
43
  try:
44
  client = InferenceClient(api_key=get_random_api_key())
45
  messages = [
46
+ {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
47
  {"role": "user", "content": edited_prompt},
48
  {"role": "assistant", "content": original_story}
49
  ]
 
67
  continuation_prompt = next_prompt
68
 
69
  messages = [
70
+ {"role": "system", "content": "Продли историю, учитывая предоставленные указания. Продливай В ТОЧНОСТИ С КОНЦА, прям с того же символа, слова, предложения. (В начале добавляй новые строки/пробел если надо для отступа). Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
71
  {"role": "user", "content": continuation_prompt},
72
  {"role": "assistant", "content": original_story}
73
  ]