Hugging Face's logo

Spaces:
geshang
/
Seg-R1-demo
Running on Zero

App Files Community
geshang commited on
Commit
8299d6d
·
verified ·
1 Parent(s): 41596ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ import gradio as gr
4
  from PIL import Image as PILImage
5
  import torchvision.transforms.functional as TF
6
  import numpy as np
 
7
  import torch
8
  from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
9
  from qwen_vl_utils import process_vision_info
 
4
  from PIL import Image as PILImage
5
  import torchvision.transforms.functional as TF
6
  import numpy as np
7
+ os.environ["CUDA_VISIBLE_DEVICES"] = "0"
8
  import torch
9
  from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
10
  from qwen_vl_utils import process_vision_info