Nymbo commited on
Commit
aa83ea8
1 Parent(s): 2c6698b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from diffusers import AutoPipelineForImage2Image, AutoPipelineForText2Image
2
  import torch
3
  import os
 
4
 
5
  try:
6
  import intel_extension_for_pytorch as ipex
@@ -68,6 +69,7 @@ t2i_pipe.set_progress_bar_config(disable=True)
68
  i2i_pipe.to(device=torch_device, dtype=torch_dtype).to(device)
69
  i2i_pipe.set_progress_bar_config(disable=True)
70
 
 
71
 
72
  def resize_crop(image, size=512):
73
  image = image.convert("RGB")
 
1
  from diffusers import AutoPipelineForImage2Image, AutoPipelineForText2Image
2
  import torch
3
  import os
4
+ import spaces
5
 
6
  try:
7
  import intel_extension_for_pytorch as ipex
 
69
  i2i_pipe.to(device=torch_device, dtype=torch_dtype).to(device)
70
  i2i_pipe.set_progress_bar_config(disable=True)
71
 
72
+ @spaces.GPU
73
 
74
  def resize_crop(image, size=512):
75
  image = image.convert("RGB")