Upload folder using huggingface_hub

README.md

@@ -1,11 +1,12 @@
----
-title: Robust-R1
-emoji: 🤖
-colorFrom: blue
-colorTo: purple
-sdk: gradio
-sdk_version: 6.1.0
-app_file: app.py
-pinned: false
-license: apache-2.0
----
+---
+title: Robust-R1
+emoji: 🤖
+colorFrom: blue
+colorTo: purple
+sdk: gradio
+sdk_version: 4.0.0
+app_file: app.py
+pinned: false
+license: apache-2.0
+---
+

app.py

@@ -5,6 +5,9 @@ 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, 
         then based on the strength of degradation, output an APPROPRIATE length for the reasoning process in <REASONING> <REASONING_END> tags,