Update app.py

app.py

@@ -13,10 +13,9 @@ MAX_SEED = np.iinfo(np.int32).max
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-torch.set_grad_enabled(False)
+
 vlm_pipe = ModularPipeline.from_pretrained("briaai/FIBO-VLM-prompt-to-JSON", trust_remote_code=True).to(device)
 pipe = BriaFiboPipeline.from_pretrained("briaai/FIBO", trust_remote_code=True, torch_dtype=dtype).to(device)
-pipe.enable_model_cpu_offload()
 
 @spaces.GPU(duration=300)
 def infer(
@@ -70,6 +69,14 @@ css = """
     margin: 0 auto;
     max-width: 768px;
 }
+#col-container button {
+    background-color: #b388ff !important;  /* light purple */
+    color: white !important;
+}
+
+#col-container .gr-accordion > div {
+    background-color: #ffd6e8 !important;  /* light pink */
+}
 """
 
 with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
@@ -85,9 +92,9 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
                 with gr.Row():
                     prompt_refine = gr.Textbox(label="Prompt")
 
-        submit_btn = gr.Button("Generate")
+        submit_btn = gr.Button("Generate", elem_id="submit-btn")
         result = gr.Image(label="output")
-        with gr.Accordion("Structured Prompt", open=False):
+        with gr.Accordion("Structured Prompt", open=False, elem_id="structured-prompt"):
             prompt_in_json = gr.JSON(label="json structured prompt")
         with gr.Accordion("Advanced Settings", open=False):
             with gr.Row():