Fix requirements

app.py

@@ -6,7 +6,10 @@ import torch
 from monai.transforms import Compose, ScaleIntensityd, SpatialPadd
 from cinema import ConvUNetR
 from pathlib import Path
-from examples.inference.segmentation_sax import plot_segmentations, plot_volume_changes
+from cinema.examples.inference.segmentation_sax import (
+    plot_segmentations,
+    plot_volume_changes,
+)
 import spaces
 
 # cache directories
@@ -112,7 +115,7 @@ with gr.Blocks(
     )
 
     with gr.Row():
-        with gr.Column(scale=0.4):
+        with gr.Column(scale=4):
             gr.Markdown("## Description")
             gr.Markdown("""
             Please adjust the settings on the right panels and click the button to run the inference.
@@ -131,7 +134,7 @@ with gr.Blocks(
             The left panel shows the segmentation of ventricles and myocardium every n time steps across all SAX slices.
             The right panel plots the ventricle and mycoardium volumes across all inference time frames.
             """)
-        with gr.Column(scale=0.3):
+        with gr.Column(scale=3):
             gr.Markdown("## Data Settings")
             image_id = gr.Slider(
                 minimum=1,
@@ -147,7 +150,7 @@ with gr.Blocks(
                 label="Choose the gap between time frames",
                 value=2,
             )
-        with gr.Column(scale=0.3):
+        with gr.Column(scale=3):
             gr.Markdown("## Model Setting")
             trained_dataset = gr.Dropdown(
                 choices=["ACDC", "M&MS", "M&MS2"],

requirements.txt

@@ -19,6 +19,5 @@ spaces==0.36.0
 timm==1.0.15
 wandb==0.19.11
 git+https://github.com/mathpluscode/CineMA#egg=cinema
-git+https://github.com/mathpluscode/CineMA#egg=examples
 --extra-index-url https://download.pytorch.org/whl/cu113
 torch==2.5.1