Update app.py

app.py

@@ -1,10 +1,12 @@
 import subprocess
+import os
+os.environ['CUDA_VISIBLE_DEVICES']='0'
 subprocess.run('pip install flash-attn==2.7.0.post2 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 subprocess.run('pip install gptqmodel==2.1.0', shell=True)
 subprocess.run('pip install numpy==1.25.0 torch==2.4.0 transformers==4.46.2 pillow==10.3.0 moviepy==1.0.3', shell=True)
+os.environ['CUDA_VISIBLE_DEVICES']='0'
 
 import spaces
-import os
 import re
 import logging
 from typing import List, Any
@@ -18,6 +20,7 @@ from PIL import Image
 from gptqmodel import GPTQModel
 
 model_name = 'AIDC-AI/Ovis2-34B-GPTQ-Int4'
+os.environ['CUDA_VISIBLE_DEVICES']='0'
 
 use_thread = False