Update api.py

api.py

@@ -1,18 +1,17 @@
-# Custom Urdu Whisper API + UI
+from transformers import WhisperProcessor, WhisperForConditionalGeneration
+import torch
+import soundfile as sf
 
-This Space runs:
-- A custom Flask API (for external apps)
-- A Gradio UI (for manual use)
-- Fully offline model (no HF hosted API)
+class UrduWhisper:
+    def __init__(self):
+        print("Loading Urdu Whisper Tiny model...")
+        self.processor = WhisperProcessor.from_pretrained("kingabzpro/whisper-tiny-urdu")
+        self.model = WhisperForConditionalGeneration.from_pretrained("kingabzpro/whisper-tiny-urdu")
+        self.model.to("cpu")
 
-## API Endpoint
-`POST /transcribe`
-
-Body:
-- `file`: audio/wav file
-
-Response:
-```json
-{
-  "text": "urdu transcription here"
-}
+    def transcribe(self, audio_file):
+        audio, sr = sf.read(audio_file)
+        inputs = self.processor(audio, sampling_rate=sr, return_tensors="pt")
+        with torch.no_grad():
+            predicted_ids = self.model.generate(inputs["input_features"])
+        return self.processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]