vibs08 commited on
Commit
0f65a27
·
verified ·
1 Parent(s): 1b6f106

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -114,7 +114,7 @@ def gen_pos_prompt(text):
114
  '''
115
 
116
  body = json.dumps({'inputText': instruction,
117
- 'textGenerationConfig': {'temperature': 0, 'topP': 0.01, 'maxTokenCount':512}})
118
  response = bedrock_runtime.invoke_model(body=body, modelId='amazon.titan-text-express-v1')
119
  pos_prompt = json.loads(response.get('body').read())['results'][0]['outputText']
120
  return pos_prompt
 
114
  '''
115
 
116
  body = json.dumps({'inputText': instruction,
117
+ 'textGenerationConfig': {'temperature': 0, 'topP': 0.01, 'maxTokenCount':1024}})
118
  response = bedrock_runtime.invoke_model(body=body, modelId='amazon.titan-text-express-v1')
119
  pos_prompt = json.loads(response.get('body').read())['results'][0]['outputText']
120
  return pos_prompt