Upload folder using huggingface_hub

app.py

@@ -1,3 +1,6 @@
+import subprocess
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
 import gradio as gr
 import os
 import torch
@@ -5,8 +8,7 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
 import html
 
-import subprocess
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
 
 sys_prompt = """First output the the types of degradations in image briefly in <TYPE> <TYPE_END> tags, 
         and then output what effects do these degradation have on the image in <INFLUENCE> <INFLUENCE_END> tags,