Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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"),
|