Hugging Face's logo

Spaces:
tobiaspires
/
ad-image-generation
Runtime error

App Files Community
tobiaspires commited on
Commit
3b14a4f
·
1 Parent(s): aa88efb

Alterando device para cpu

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image, ImageOps
6
  import PIL
7
 
8
  # cuda cpu
9
- device_name = 'cuda'
10
  device = torch.device(device_name)
11
 
12
  processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined")
@@ -82,4 +82,4 @@ with gr.Blocks() as demo:
82
  )
83
 
84
  demo.queue(concurrency_count=1)
85
- demo.launch()
 
6
  import PIL
7
 
8
  # cuda cpu
9
+ device_name = 'cpu'
10
  device = torch.device(device_name)
11
 
12
  processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined")
 
82
  )
83
 
84
  demo.queue(concurrency_count=1)
85
+ demo.launch()