rollback 2 days

app.py

@@ -5,36 +5,32 @@ from transformers import pipeline
 
 MODEL_NAME = "palli23/whisper-small-sam_spjall"
 
-# ← Only these 3 tiny changes
-@spaces.GPU(duration=120)           # was 60 → give it breathing room
+@spaces.GPU(duration=120)   # nóg fyrir 3 mín hljóð
 def transcribe_3min(audio_path):
     if not audio_path:
         return "Hladdu upp hljóðskrá"
-
-    # ← This is the ONLY line we touch inside the function
+    
+    # Whisper pipeline með chunking – ZeroGPU öruggt
     pipe = pipeline(
         "automatic-speech-recognition",
         model=MODEL_NAME,
         device=0,
         token=os.getenv("HF_TOKEN"),
-        torch_dtype="float16",                    # ← NEW: FP16 instead of FP32
-        model_kwargs={"use_flash_attention_2": True}  # ← NEW: FlashAttention-2 (huge speedup on T4)
+        torch_dtype="float16",    # ← NEW: FP16 instead of FP32
+        # ← REMOVED: use_flash_attention_2 (causes TypeError on this model)
     )
-
+    
     result = pipe(
         audio_path,
-        chunk_length_s=30,
-        stride_length_s=(6, 0),
+        chunk_length_s=30,           # 30 sek chunkar
+        stride_length_s=(6, 0),      # 6 sek overlap
         return_timestamps=False,
-        batch_size=8          # ← now actually works because of FP16 + FlashAttn
+        batch_size=8
     )
     
     return result["text"]
 
-
-# ──────────────────────────────────────────────
-# Rest of your code 100% unchanged
-# ──────────────────────────────────────────────
+# Interface
 with gr.Blocks(title="Íslenskt ASR – 3 mín") as demo:
     gr.Markdown("# Íslenskt ASR – 3 mínútur")
     gr.Markdown("**Whisper · Very low WER · 0.5-5minute audio transcribe á ZeroGPU**")