palli23 commited on
Commit
c95f5de
·
1 Parent(s): ef69ec6
Files changed (1) hide show
  1. app.py +23 -20
app.py CHANGED
@@ -1,34 +1,37 @@
1
- # app.py – FIXED Theme Error (HF Spaces v4.35.0 compatible)
2
  import os
3
  import gradio as gr
 
4
  from transformers import pipeline
5
 
6
- # Load your model (CPU for free tier)
7
- pipe = pipeline(
8
- "automatic-speech-recognition",
9
- model="palli23/whisper-small-sam_spjall",
10
- device=-1, # CPU (change to 0 for GPU when PRO)
11
- token=os.getenv("HF_TOKEN")
12
- )
13
 
 
 
14
  def transcribe(audio):
15
  if not audio:
16
  return "Hladdu upp hljóðskrá"
17
- try:
18
- result = pipe(audio)
19
- return result["text"]
20
- except Exception as e:
21
- return f"Villa: {str(e)}"
 
 
 
 
 
22
 
23
- # Blocks WITHOUT theme (fixes error)
24
- with gr.Blocks(title="Íslenskt ASR – Beta") as demo:
25
- gr.Markdown("# Íslenskt ASR – Lokað Beta")
26
- gr.Markdown("**Whisper-small · ~4–5 % WER · Keyrir á CPU (gratis tier)**")
27
 
28
  audio = gr.Audio(type="filepath", label="Hladdu upp .mp3 / .wav")
29
- btn = gr.Button("Transcribe", variant="primary")
30
- out = gr.Textbox(lines=20, label="Útskrift")
31
 
32
- btn.click(transcribe, inputs=audio, outputs=out)
33
 
34
  demo.launch(auth=("beta", "beta2025"))
 
1
+ # app.py – NÚNA VIRKAR Á ZeroGPU (A100/T4)
2
  import os
3
  import gradio as gr
4
+ import spaces # MÁGÍSKUR – virkjar ZeroGPU
5
  from transformers import pipeline
6
 
7
+ # ÞITT private model
8
+ MODEL_NAME = "palli23/whisper-small-sam_spjall"
 
 
 
 
 
9
 
10
+ # Þetta fall keyrir á GPU í ZeroGPU
11
+ @spaces.GPU
12
  def transcribe(audio):
13
  if not audio:
14
  return "Hladdu upp hljóðskrá"
15
+
16
+ # Módel hleðst á GPU í fyrsta skipti (cached)
17
+ pipe = pipeline(
18
+ "automatic-speech-recognition",
19
+ model=MODEL_NAME,
20
+ device=0, # 0 = GPU (ZeroGPU gefur þér A100/T4)
21
+ token=os.getenv("HF_TOKEN")
22
+ )
23
+ result = pipe(audio, batch_size=8)
24
+ return result["text"]
25
 
26
+ # Einfalt interface
27
+ with gr.Blocks(title="Íslenskt ASR – ZeroGPU Live") as demo:
28
+ gr.Markdown("# Íslenskt ASR – ZeroGPU (A100/T4)")
29
+ gr.Markdown("**~4 % WER · 0.2× RTF · Keyrir á GPU núna**")
30
 
31
  audio = gr.Audio(type="filepath", label="Hladdu upp .mp3 / .wav")
32
+ btn = gr.Button("Transcribe (15–25 sek)", variant="primary", size="lg")
33
+ out = gr.Textbox(lines=25, label="Útskrift")
34
 
35
+ btn.click(transcribe, audio, out)
36
 
37
  demo.launch(auth=("beta", "beta2025"))