Hugging Face's logo

Spaces:
alimboff
/
nllb-200-kbd
Paused

App Files Community
alimboff commited on
Commit
8d4dfa0
·
verified ·
1 Parent(s): f98f8b3

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import NllbTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
- model = AutoModelForSeq2SeqLM.from_pretrained('alimboff/nllb-200-kbd').cuda()
5
  tokenizer = NllbTokenizer.from_pretrained('alimboff/nllb-200-kbd')
6
 
7
  def fix_tokenizer(tokenizer, new_lang='kbd_Cyrl'):
 
1
  from transformers import NllbTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
+ model = AutoModelForSeq2SeqLM.from_pretrained('alimboff/nllb-200-kbd')#.cpu()
5
  tokenizer = NllbTokenizer.from_pretrained('alimboff/nllb-200-kbd')
6
 
7
  def fix_tokenizer(tokenizer, new_lang='kbd_Cyrl'):