Spaces:
Running
Running
Update app.py
Browse files
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/
|
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/
|
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:
|