Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 from PIL import Image as PILImage
 import torchvision.transforms.functional as TF
 import numpy as np
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 import torch
 from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
 from qwen_vl_utils import process_vision_info