Update app.py

app.py

@@ -142,7 +142,7 @@ class ModelWrapper:
                 )
         return prompt
     
-    @spaces.GPU
+    @spaces.GPU(duration=240)
     def get_response(self, input, temperature=0.7, top_k=20, top_p=0.8, max_new_tokens=256,
                      enable_thinking=True, system_prompt=SYSTEM_PROMPT):
         print("Generating response...")
@@ -233,6 +233,8 @@ Agent: Oh sure, Mr. Choose-Your-Own-Math-Adventure, that's the best solution I'v
                 "yueliu1999/GuardReasoner-8B",
                 "allenai/wildguard",
                 "Qwen/Qwen3-0.6B",
+                "tomg-group-umd/DynaGuard-4B",
+                "tomg-group-umd/DynaGuard-1.7B",
             ],
             label="Select Model",
             value="tomg-group-umd/DynaGuard-8B",