MohamedRashad commited on
Commit
3ab33fa
·
verified ·
1 Parent(s): 5d44d4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,3 @@
1
- import os
2
- os.system("pip install -U flash-attn")
3
-
4
  from transformers import (
5
  NougatProcessor,
6
  VisionEncoderDecoderModel,
@@ -14,6 +11,9 @@ import spaces
14
  from threading import Thread
15
  from gradio_pdf import PDF
16
 
 
 
 
17
  models_supported = {
18
  "arabic-small-nougat": [
19
  NougatProcessor.from_pretrained("MohamedRashad/arabic-small-nougat"),
 
 
 
 
1
  from transformers import (
2
  NougatProcessor,
3
  VisionEncoderDecoderModel,
 
11
  from threading import Thread
12
  from gradio_pdf import PDF
13
 
14
+ import os
15
+ os.system("pip install -U flash-attn")
16
+
17
  models_supported = {
18
  "arabic-small-nougat": [
19
  NougatProcessor.from_pretrained("MohamedRashad/arabic-small-nougat"),