adi-123 commited on
Commit
7fd9800
·
1 Parent(s): e15f81c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -35,11 +35,9 @@ def txt2story(img_text):
35
 
36
  print("Preparing message sequences for interaction...")
37
  messages = [
38
- {"role": "system", "content": "Once upon a time..."},
39
  {"role": "user", "content": img_text, "temperature": 1},
40
- {"role": "system",
41
- "content": content_prompt,
42
- "temperature": 0.7},
43
  ]
44
 
45
  print("Generating story completion using the AI model...")
 
35
 
36
  print("Preparing message sequences for interaction...")
37
  messages = [
38
+ {"role": "system", "content": "You are an expert short story writer who specialises in ending stories positively."},
39
  {"role": "user", "content": img_text, "temperature": 1},
40
+ {"role": "system", "content": content_prompt, "temperature": 0.7},
 
 
41
  ]
42
 
43
  print("Generating story completion using the AI model...")