Update app.py

app.py

@@ -106,12 +106,12 @@ def generate_embeddings(texts: Union[str, List[str]]) -> Union[List[float], List
                             embedding = (sum_embeddings / sum_mask).squeeze().cpu().numpy()
                         else:
                             # Simple mean pooling without attention mask
-                        embedding = outputs.last_hidden_state.mean(dim=1).squeeze().cpu().numpy()
+                            embedding = outputs.last_hidden_state.mean(dim=1).squeeze().cpu().numpy()
                     else:
                         # Fallback to pooled output if available
                         embedding = outputs.pooler_output.squeeze().cpu().numpy()
                     
-                        embeddings.append(embedding.tolist())
+                    embeddings.append(embedding.tolist())
                         
             except Exception as e:
                 logger.error(f"Error generating embedding for text: {str(e)}")