zhiweili commited on
Commit
286713d
·
1 Parent(s): 115c9f4

import spaces

Browse files
Files changed (1) hide show
  1. app_upscale.py +1 -0
app_upscale.py CHANGED
@@ -4,6 +4,7 @@ from io import BytesIO
4
  from diffusers import StableDiffusionUpscalePipeline
5
  import torch
6
  import gradio as gr
 
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
 
4
  from diffusers import StableDiffusionUpscalePipeline
5
  import torch
6
  import gradio as gr
7
+ import spaces
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10