Hugging Face's logo

Spaces:
amirza
/
draw_me_a_sheep_heb
Runtime error

App Files Community
Amir Zait commited on
Commit
f7895ed
•
1 Parent(s): 2d31a01

fixed model

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -15,9 +15,10 @@ from vqgan_jax.modeling_flax_vqgan import VQModel
15
 
16
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
 
18
- api_token = os.getenv("API_TOKEN")
19
- asr_processor = AutoProcessor.from_pretrained("imvladikon/wav2vec2-xls-r-300m-hebrew")
20
- asr_model = AutoModelForCTC.from_pretrained("imvladikon/wav2vec2-xls-r-300m-hebrew")
 
21
 
22
  he_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-he-en")
23
 
 
15
 
16
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
 
18
+ #asr_processor = AutoProcessor.from_pretrained("imvladikon/wav2vec2-xls-r-300m-hebrew")
19
+ #asr_model = AutoModelForCTC.from_pretrained("imvladikon/wav2vec2-xls-r-300m-hebrew")
20
+ asr_processor = AutoProcessor.from_pretrained("imvladikon/wav2vec2-xls-r-1b-hebrew")
21
+ asr_model = AutoModelForCTC.from_pretrained("imvladikon/wav2vec2-xls-r-1b-hebrew")
22
 
23
  he_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-he-en")
24