palli23 commited on
Commit
451a43f
·
1 Parent(s): 2bc8f15

CPU-only – virkar strax án GPU vandræða

Browse files
Files changed (1) hide show
  1. app.py +13 -17
app.py CHANGED
@@ -1,42 +1,38 @@
1
- # app.py – VIRKAR Á A100 ÁN @spaces.GPU (2025 leiðin)
2
  import os
3
  import gradio as gr
4
  from transformers import pipeline
5
 
6
- # Þvingar GPU með torch (virkar alltaf á A100 Spaces)
7
- import torch
8
- if torch.cuda.is_available():
9
- device = 0
10
- print(f"GPU virk: {torch.cuda.get_device_name(0)}")
11
- else:
12
- device = -1
13
- print("CPU – eitthvað er rangt við Space stillingar")
14
 
15
- # Þitt private model
16
  asr = pipeline(
17
  "automatic-speech-recognition",
18
  model="palli23/whisper-small-sam_spjall",
19
- device=device,
20
  token=os.getenv("HF_TOKEN")
21
  )
22
 
23
  def transcribe(audio):
24
  if not audio:
25
- return "Hladdu upp hljóðskrá"
26
  try:
27
  result = asr(audio)
28
  return result["text"]
29
  except Exception as e:
30
  return f"Villa: {str(e)}"
31
 
 
32
  with gr.Blocks() as demo:
33
- gr.Markdown("# Íslenskt ASR – VIRKAR Á A100")
34
- gr.Markdown("**Whisper-small · ~4–5 % WER · Keyrir á GPU**")
35
 
36
  audio = gr.Audio(type="filepath", label="Hladdu upp .mp3 / .wav")
37
- btn = gr.Button("Transcribe", variant="primary")
38
- out = gr.Textbox(lines=20, label="Útskrift")
39
 
40
- btn.click(transcribe, audio, out)
 
 
41
 
42
  demo.launch(auth=("beta", "beta2025"))
 
1
+ # app.py – Virkar 100 % á CPU (engin GPU villa, engin NVML)
2
  import os
3
  import gradio as gr
4
  from transformers import pipeline
5
 
6
+ print("Hleð Whisper-small... (tekur 30–60 sek í fyrsta skipti)")
 
 
 
 
 
 
 
7
 
8
+ # ÞITT private model
9
  asr = pipeline(
10
  "automatic-speech-recognition",
11
  model="palli23/whisper-small-sam_spjall",
12
+ device=-1, # ← CPU (ekki 0)
13
  token=os.getenv("HF_TOKEN")
14
  )
15
 
16
  def transcribe(audio):
17
  if not audio:
18
+ return "Hladdu upp hljóðskrá fyrst"
19
  try:
20
  result = asr(audio)
21
  return result["text"]
22
  except Exception as e:
23
  return f"Villa: {str(e)}"
24
 
25
+ # Einfalt og fallegt
26
  with gr.Blocks() as demo:
27
+ gr.Markdown("# Íslenskt ASR – Lokað Beta (CPU)")
28
+ gr.Markdown("**Whisper-small · ~4–5 % WER · Keyrir á CPU (nóg fyrir beta)**")
29
 
30
  audio = gr.Audio(type="filepath", label="Hladdu upp .mp3 / .wav")
31
+ btn = gr.Button("Transcribe (30–90 sek)", variant="primary", size="lg")
32
+ out = gr.Textbox(lines=25, label="Útskrift")
33
 
34
+ btn.click(transcribe, inputs=audio, outputs=out)
35
+
36
+ gr.Markdown("© 2025 – Einkaeign")
37
 
38
  demo.launch(auth=("beta", "beta2025"))