Rooni commited on
Commit
9722a9c
·
verified ·
1 Parent(s): fc77950

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,7 +22,7 @@ except requests.exceptions.RequestException as e:
22
 
23
  def generate_story(prompt, style):
24
  try:
25
- client = InferenceClient(api_key=get_random_api_key())
26
 
27
  messages = [
28
  {"role": "system", "content": f"Напиши хорошую историю в стиле '{style}'. Подробную, понятную, человечную (с душой), уникальную. Не обязательно делать концовку, можно только начало длинной истории. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
@@ -47,7 +47,7 @@ def edit_story(original_story="", edited_prompt=""):
47
  return
48
 
49
  try:
50
- client = InferenceClient(api_key=get_random_api_key())
51
  messages = [
52
  {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
53
  {"role": "user", "content": edited_prompt},
@@ -68,7 +68,7 @@ def next_story_func(original_story="", next_prompt="", continuation_type="Про
68
  return
69
 
70
  try:
71
- client = InferenceClient(api_key=get_random_api_key())
72
  if continuation_type == "Продолжение следует...":
73
  continuation_prompt = "Продолжи, подведя к тому, что продолжение следует."
74
  elif continuation_type == "Концовка":
 
22
 
23
  def generate_story(prompt, style):
24
  try:
25
+ client = InferenceClient(provider="together", api_key=get_random_api_key())
26
 
27
  messages = [
28
  {"role": "system", "content": f"Напиши хорошую историю в стиле '{style}'. Подробную, понятную, человечную (с душой), уникальную. Не обязательно делать концовку, можно только начало длинной истории. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
 
47
  return
48
 
49
  try:
50
+ client = InferenceClient(provider="together", api_key=get_random_api_key())
51
  messages = [
52
  {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."},
53
  {"role": "user", "content": edited_prompt},
 
68
  return
69
 
70
  try:
71
+ client = InferenceClient(provider="together", api_key=get_random_api_key())
72
  if continuation_type == "Продолжение следует...":
73
  continuation_prompt = "Продолжи, подведя к тому, что продолжение следует."
74
  elif continuation_type == "Концовка":