Update fast_api.py

fast_api.py

@@ -14,23 +14,23 @@ def load_whisper_model(model_path, device='cpu'):
         whisper.model.ModelDimensions(
             n_mels=80,
             n_audio_ctx=1500,
-            n_audio_state=192,  # Adjusted for Tiny
-            n_audio_head=3,     # Adjusted for Tiny
-            n_audio_layer=2,    # Adjusted for Tiny
+            n_audio_state=384,  # Adjusted to match checkpoint
+            n_audio_head=3,
+            n_audio_layer=2,
             n_vocab=51865,
             n_text_ctx=448,
-            n_text_state=192,   # Adjusted for Tiny
-            n_text_head=3,      # Adjusted for Tiny
-            n_text_layer=2      # Adjusted for Tiny
+            n_text_state=384,   # Adjusted to match checkpoint
+            n_text_head=3,
+            n_text_layer=2
         )
     )
-    
+
     # Load state dict
     state_dict = torch.load(model_path, map_location=device, weights_only=True)
     model.load_state_dict(state_dict)
     model.eval()
     return model
-    
+
 # Load the saved Word2Vec model
 word2vec_model = KeyedVectors.load("word2vec-google-news-300.model")
 model = load_whisper_model("whisper_tiny_model.pt")