Rooni commited on
Commit
2c0b5a6
·
verified ·
1 Parent(s): b8ec7a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,7 +39,7 @@ def generate_story(prompt, style):
39
  {"role": "user", "content": prompt}
40
  ]
41
 
42
- completion = client.chat.completions.create(model="Qwen/QwQ-32B-Preview", messages=messages, max_tokens=1000)
43
  story = completion.choices[0].message.content
44
  return story
45
  except Exception as e:
@@ -56,7 +56,7 @@ def edit_story(original_story, edited_prompt):
56
  {"role": "assistant", "content": original_story}
57
  ]
58
 
59
- completion = client.chat.completions.create(model="Qwen/QwQ-32B-Preview", messages=messages, max_tokens=1000)
60
  edited_story = completion.choices[0].message.content
61
  return edited_story
62
  except Exception as e:
 
39
  {"role": "user", "content": prompt}
40
  ]
41
 
42
+ completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, max_tokens=1000)
43
  story = completion.choices[0].message.content
44
  return story
45
  except Exception as e:
 
56
  {"role": "assistant", "content": original_story}
57
  ]
58
 
59
+ completion = client.chat.completions.create(model="Qwen/Qwen2.5-Coder-32B-Instruct", messages=messages, max_tokens=1000)
60
  edited_story = completion.choices[0].message.content
61
  return edited_story
62
  except Exception as e: