kingabzpro commited on
Commit
a313388
·
verified ·
1 Parent(s): ec69d68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,7 +26,7 @@ model_id = "kingabzpro/whisper-base-urdu-full"
26
 
27
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
28
  model_id,
29
- torch_dtype=torch.float32,
30
  use_safetensors=True,
31
  )
32
 
@@ -93,7 +93,7 @@ demo = gr.Interface(
93
  description=description,
94
  examples=examples,
95
  allow_flagging="never",
96
- theme='shivi/calm_seafoam',
97
  )
98
 
99
  if __name__ == "__main__":
 
26
 
27
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
28
  model_id,
29
+ torch_dtype=torch.float16,
30
  use_safetensors=True,
31
  )
32
 
 
93
  description=description,
94
  examples=examples,
95
  allow_flagging="never",
96
+ theme=gr.themes.Soft(),
97
  )
98
 
99
  if __name__ == "__main__":