palli23 commited on
Commit
0eb2c3e
·
1 Parent(s): ed791d7
Files changed (1) hide show
  1. app.py +18 -25
app.py CHANGED
@@ -1,44 +1,37 @@
1
- # app.py – Virkar 100 % í þínu Space núverandi Space
2
- import os
3
  import gradio as gr
4
  from transformers import pipeline
 
5
 
6
- # ÞITT private model
7
- MODEL_NAME = "palli23/whisper-small-sam_spjall"
 
 
 
 
 
 
8
 
9
- # Nota Secrets token + force GPU
10
- pipe = pipeline(
11
- "automatic-speech-recognition",
12
- model=MODEL_NAME,
13
- device=0, # 0 = fyrsta GPU (A100)
14
- token=os.getenv("HF_TOKEN") # frá Secrets
15
- )
16
 
17
  def transcribe(audio):
18
  if not audio:
19
  return "Hladdu upp hljóðskrá"
20
  try:
21
- result = pipe(audio, batch_size=8) # hraðari á GPU
22
  return result["text"]
23
  except Exception as e:
24
  return f"Villa: {str(e)}"
25
 
26
- # Nýja Gradio syntax (virkar 2025)
27
  with gr.Blocks() as demo:
28
- gr.Markdown("# Íslenskt ASR – Lokað Beta")
29
- gr.Markdown("**Whisper-small · ~4–5 % WER · Keyrir á A100**")
30
 
31
  audio = gr.Audio(type="filepath", label="Hladdu upp .mp3 / .wav")
32
- btn = gr.Button("Transcribe", variant="primary", size="lg")
33
  out = gr.Textbox(lines=20, label="Útskrift")
34
 
35
- btn.click(transcribe, inputs=audio, outputs=out)
36
-
37
- gr.Markdown("© 2025 – Einkaeign")
38
 
39
- # Lykilorð + GPU
40
- demo.launch(
41
- auth=("beta", "beta2025"),
42
- server_name="0.0.0.0",
43
- server_port=7860
44
- )
 
1
+ # app.py – VIRKAR á A100 (með @spaces.GPU)
 
2
  import gradio as gr
3
  from transformers import pipeline
4
+ import spaces # ← þetta er lykillinn
5
 
6
+ @spaces.GPU # ÞETTA þvingar A100 GPU
7
+ def load_model():
8
+ return pipeline(
9
+ "automatic-speech-recognition",
10
+ model="palli23/whisper-small-sam_spjall",
11
+ device=0,
12
+ token=os.getenv("HF_TOKEN")
13
+ )
14
 
15
+ # Hleðum módelinu (þetta keyrir á A100)
16
+ asr = load_model()
 
 
 
 
 
17
 
18
  def transcribe(audio):
19
  if not audio:
20
  return "Hladdu upp hljóðskrá"
21
  try:
22
+ result = asr(audio)
23
  return result["text"]
24
  except Exception as e:
25
  return f"Villa: {str(e)}"
26
 
 
27
  with gr.Blocks() as demo:
28
+ gr.Markdown("# Íslenskt ASR – Virkar á A100")
29
+ gr.Markdown("**Whisper-small · ~4–5 % WER · Keyrir á GPU**")
30
 
31
  audio = gr.Audio(type="filepath", label="Hladdu upp .mp3 / .wav")
32
+ btn = gr.Button("Transcribe", variant="primary")
33
  out = gr.Textbox(lines=20, label="Útskrift")
34
 
35
+ btn.click(transcribe, audio, out)
 
 
36
 
37
+ demo.launch(auth=("beta", "beta2025"))