Restore GPU version - remove CPU optimizations and restore GPU-compatible dependencies

app.py

@@ -39,7 +39,7 @@ temp_generation_config = GenerationConfig(
 # ----------------------------
 # YOLO MODEL LOADING
 # ----------------------------
-yolo_model = YOLO("best.pt")  # replace with your segmentation weights
+yolo_model = YOLO("/kaggle/working/Brain-Tumor-Segmentation/best.pt")  # replace with your segmentation weights
 
 def inference(image_path: str):
     """Runs YOLO segmentation on an image and returns the annotated image."""
@@ -175,7 +175,15 @@ Original Answer:
 # ----------------------------
 # GRADIO UI
 # ----------------------------
-with gr.Blocks(title="Brain MRI QA") as demo:
+css = """
+.scrollable-column {
+    max-height: 80vh;
+    overflow-y: auto;
+    padding-right: 10px;
+}
+"""
+
+with gr.Blocks(title="Brain MRI QA", css=css) as demo:
     with gr.Row():
         # First column: input image and result image side by side
         with gr.Column():
@@ -192,7 +200,7 @@ with gr.Blocks(title="Brain MRI QA") as demo:
                 clear_btn = gr.Button("Clear")
 
         # Second column: LLM answer output
-        with gr.Column():
+        with gr.Column(elem_classes="scrollable-column"):
             llm_output = gr.Markdown(label="Detailed LLM Answer")
     submit_btn.click(
         fn=process_pipeline,