minor update

app.py

@@ -296,8 +296,8 @@ def atlas_slice_prediction(user_section, axis = 'coronal'):
   user_section = square_padding(user_section, 224)
   user_section = (user_section - np.min(user_section))/((np.max(user_section) - np.min(user_section)))
   print("Loading model")
-  atlas_embeddings = np.load(f"registration/template_id_checkpoints/atlas_embeddings_{axis}.npy")
-  atlas_labels = np.load(f"registration/template_id_checkpoints/atlas_labels_{axis}.npy")
+  atlas_embeddings = np.load(f"registration/atlas_embeddings_{axis}.npy")
+  atlas_labels = np.load(f"registration/atlas_labels_{axis}.npy")
   idx = embeddings_classifier(user_section, atlas_embeddings,atlas_labels)
 
   return idx
@@ -475,7 +475,7 @@ with gr.Blocks() as demo:
     gr.Markdown("### Step 1: Upload your sample and choose type")
     with gr.Row():
         nifti_file = gr.File(label="File Upload")
-        with gr.Row():
+        with gr.Column():
           sample_type = gr.Dropdown(choices=["CCF registered Sample", "Custom Sample"], value="CCF registered Sample", label="Sample Type")
           data_type = gr.Radio(choices=["2D", "3D"], value="3D", label="Data Type")