Update app.py

app.py

@@ -189,30 +189,31 @@ with gr.Blocks(title="Cosmos-Reason1", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# Cosmos-Reason1")
     gr.Markdown("Upload an image or video and ask a question about it.")
     
-    with gr.Accordion("Model Configuration", open=False):
-        attention_mode = gr.Dropdown(
-            choices=["sdpa", "xformers", "flash_attention_2"],
-            value=config["attention_mode"],
-            label="Attention Mode"
-        )
-        torch_dtype = gr.Dropdown(
-            choices=["float16", "bfloat16", "float32"],
-            value=config["torch_dtype"],
-            label="Torch Data Type"
-        )
-        device_map = gr.Dropdown(
-            choices=["auto", "cuda", "cpu"],
-            value=config["device_map"],
-            label="Device Map"
-        )
-        config_btn = gr.Button("Apply Configuration")
-        config_msg = gr.Markdown()
+    # with gr.Accordion("Model Configuration", open=False):
+    #     attention_mode = gr.Dropdown(
+    #         choices=["sdpa", "xformers", "flash_attention_2"],
+    #         value=config["attention_mode"],
+    #         label="Attention Mode"
+    #     )
+    #     torch_dtype = gr.Dropdown(
+    #         choices=["float16", "bfloat16", "float32"],
+    #         value=config["torch_dtype"],
+    #         label="Torch Data Type"
+    #     )
+    #     device_map = gr.Dropdown(
+    #         choices=["auto", "cuda", "cpu"],
+    #         value=config["device_map"],
+    #         label="Device Map"
+    #     )
+    #     config_btn = gr.Button("Apply Configuration")
+    #     config_msg = gr.Markdown()
         
-        config_btn.click(
-            fn=apply_config_changes,
-            inputs=[attention_mode, torch_dtype, device_map],
-            outputs=config_msg
-        )
+    #     config_btn.click(
+    #         fn=apply_config_changes,
+    #         inputs=[attention_mode, torch_dtype, device_map],
+    #         outputs=config_msg
+    #     )
+    
     
     with gr.Row():
         with gr.Column():