Update app.py

app.py

@@ -18,7 +18,15 @@
 # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
 # -*- coding:utf-8 -*-
-
+import os
+import subprocess
+
+# Install flash attention
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
 import base64
 from io import BytesIO
 import spaces
@@ -46,7 +54,7 @@ from app_modules.conversation import SeparatorStyle
 
 def load_models():
     models = {
-        "DeepSeek-VL 7B": "deepseek-ai/deepseek-vl-7b-chat",
+        "DeepSeek-VL 7B": ""OpenGVLab/InternVL-Chat-V1-5",
     }
 
     for model_name in models: