Update model.py

model.py

@@ -164,11 +164,11 @@ if __name__ == '__main__':
     model = AutoModel.from_pretrained("INF5")
 
     # Step 1: Extract speaker embedding from reference audio + text
-    speaker_embedding, _ = extract_speaker_embedding(
+    speaker_embedding = extract_speaker_embedding(
         "prompts/PAN_F_HAPPY_00001.wav",
         "ਭਹੰਪੀ ਵਿੱਚ ਸਮਾਰਕਾਂ ਦੇ ਭਵਨ ਨਿਰਮਾਣ ਕਲਾ ਦੇ ਵੇਰਵੇ ਗੁੰਝਲਦਾਰ ਅਤੇ ਹੈਰਾਨ ਕਰਨ ਵਾਲੇ ਹਨ, ਜੋ ਮੈਨੂੰ ਖੁਸ਼ ਕਰਦੇ  ਹਨ।"
     )
-    np.save("speaker_embedding.npy", speaker_embedding.cpu().numpy())
+    nnp.save("speaker_embedding.npy", speaker_embedding)
 
     # Step 2: Load saved embedding (simulate reuse)
     loaded_embedding = np.load("speaker_embedding.npy")