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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -3,13 +3,14 @@ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
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
 
 
3
  import pytesseract
4
 
5
 
6
+ # Use a pipeline as a high-level helper
7
+ from transformers import pipeline
8
+
9
+ messages = [
10
+ {"role": "user", "content": "Who are you?"},
11
+ ]
12
+ pipe = pipeline("text-generation", model="sambanovasystems/SambaLingo-Arabic-Chat")
13
+ pipe(messages)
14
 
15
 
16