samvb1002 commited on
Commit
3faa7c6
·
verified ·
1 Parent(s): eba41fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -3,10 +3,15 @@ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
3
  import pytesseract
4
 
5
 
6
-
7
  # Load model directly
8
- tokenizer = AutoTokenizer.from_pretrained("aubmindlab/bert-base-arabic")
9
- model = AutoModelForCausalLM.from_pretrained("aubmindlab/bert-base-arabic")
 
 
 
 
 
 
10
 
11
  # Chat function
12
  def chat_fn(history, user_input):
 
3
  import pytesseract
4
 
5
 
 
6
  # Load model directly
7
+ from transformers import AutoProcessor, AutoModelForCTC
8
+
9
+ processor = AutoProcessor.from_pretrained("jonatasgrosman/wav2vec2-large-xlsr-53-arabic")
10
+ model = AutoModelForCTC.from_pretrained("jonatasgrosman/wav2vec2-large-xlsr-53-arabic")
11
+
12
+
13
+
14
+
15
 
16
  # Chat function
17
  def chat_fn(history, user_input):