Update app.py

app.py

@@ -137,13 +137,13 @@ def generate_image_from_text(pos_prompt, seed):
                             'negativeText': neg_prompt},
       'imageGenerationConfig': {"cfgScale":8,
                                 "seed":int(seed),
-                                "width":512,
-                                "height":512,
+                                "width":1024,
+                                "height":1024,
                                 "numberOfImages":1
                                 }
   }
   request_body = json.dumps(parameters)
-  response = bedrock_runtime.invoke_model(body=request_body, modelId='amazon.titan-image-generator-v1')
+  response = bedrock_runtime.invoke_model(body=request_body, modelId='amazon.titan-image-generator-v2:0')
   response_body = json.loads(response.get('body').read())
   base64_image_data = base64.b64decode(response_body['images'][0])