[FIX]UI

app.py

@@ -76,12 +76,12 @@ class LateFusionModel(nn.Module):
         self.text_model = text_model
         # MLP fusion layer (matching saved model structure)
         # Structure: Linear(2, hidden) -> ReLU -> Dropout -> Linear(hidden, 1)
-        hidden_dim = 64  # Adjust if needed based on saved model
+        hidden_dim = 16  # Matching saved model: [16, 2] -> [16] -> [1, 16]
         self.fusion_mlp = nn.Sequential(
-            nn.Linear(2, hidden_dim),  # layer 0
+            nn.Linear(2, hidden_dim),  # layer 0: [16, 2]
             nn.ReLU(),                 # layer 1 (no params)
             nn.Dropout(0.1),           # layer 2 (no params)
-            nn.Linear(hidden_dim, 1)   # layer 3
+            nn.Linear(hidden_dim, 1)   # layer 3: [1, 16]
         )
 
     def forward(self, images, input_ids, attention_mask):