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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ import spaces
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": [
 
11
  from threading import Thread
12
  from gradio_pdf import PDF
13
 
14
+ import subprocess
15
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
16
 
17
  models_supported = {
18
  "arabic-small-nougat": [