Update app.py

app.py

@@ -99,7 +99,7 @@ async def greet(product,description):
     # config = RailsConfig.from_path("nemo")
     YAML_CONFIG = """
     models:
-      - type: main
+      - type: shield
         engine: groq
         model: llama-3.2-11b-text-preview
         
@@ -110,7 +110,7 @@ async def greet(product,description):
     rails:
       input:
         flows:
-          - content safety check input $model=main
+          - content safety check input $model=shield
 
     output:
         flows: