Update app.py

app.py

@@ -26,7 +26,7 @@ from rex_omni.tasks import KEYPOINT_CONFIGS, TASK_CONFIGS, get_task_config
 
 
 def parse_args():
-    parser = argparse.ArgumentParser(description="Rex Omni Gradio Demo")
+    parser = argparse.ArgumentParser(description="Rex-Omni Gradio Demo")
     parser.add_argument(
         "--model_path",
         default="IDEA-Research/Rex-Omni",
@@ -35,14 +35,14 @@ def parse_args():
     parser.add_argument(
         "--backend",
         type=str,
-        default="transformers",
+        default="vllm",
         choices=["transformers", "vllm"],
         help="Backend to use for inference",
     )
     parser.add_argument("--temperature", type=float, default=0.0)
     parser.add_argument("--top_p", type=float, default=0.05)
     parser.add_argument("--top_k", type=int, default=1)
-    parser.add_argument("--max_tokens", type=int, default=2048)
+    parser.add_argument("--max_tokens", type=int, default=4096)
     parser.add_argument("--repetition_penalty", type=float, default=1.05)
     parser.add_argument("--min_pixels", type=int, default=16 * 28 * 28)
     parser.add_argument("--max_pixels", type=int, default=2560 * 28 * 28)