Leeps commited on
Commit
1cb7498
·
1 Parent(s): 42c1378

Import spaces before CUDA libraries

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -1,13 +1,6 @@
1
  import gc
2
  from functools import lru_cache
3
 
4
- import gradio as gr
5
- import numpy as np
6
- import torch
7
- from PIL import Image, ImageDraw
8
-
9
- from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, StableDiffusionPipeline
10
-
11
  try:
12
  import spaces
13
  except ImportError:
@@ -24,6 +17,13 @@ except ImportError:
24
 
25
  spaces = _SpacesFallback()
26
 
 
 
 
 
 
 
 
27
 
28
  APP_TITLE = "Stable Diffusion Equation Playground"
29
  DEFAULT_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 
1
  import gc
2
  from functools import lru_cache
3
 
 
 
 
 
 
 
 
4
  try:
5
  import spaces
6
  except ImportError:
 
17
 
18
  spaces = _SpacesFallback()
19
 
20
+ import gradio as gr
21
+ import numpy as np
22
+ import torch
23
+ from PIL import Image, ImageDraw
24
+
25
+ from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, StableDiffusionPipeline
26
+
27
 
28
  APP_TITLE = "Stable Diffusion Equation Playground"
29
  DEFAULT_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"