Hzqhssn commited on
Commit
b6addff
·
1 Parent(s): 1c4b8e5
Files changed (2) hide show
  1. app.py +3 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,6 +1,8 @@
 
 
1
  import torch
2
  from diffusers import StableDiffusionPipeline
3
- import gradio as gr
4
 
5
  model_id = "CompVis/stable-diffusion-v1-4"
6
  device = "cuda"
 
1
+ import spaces
2
+ import gradio as gr
3
  import torch
4
  from diffusers import StableDiffusionPipeline
5
+
6
 
7
  model_id = "CompVis/stable-diffusion-v1-4"
8
  device = "cuda"
requirements.txt CHANGED
@@ -6,5 +6,6 @@ open_clip_torch
6
  diffusers
7
  transformers
8
  scipy
 
9
  # This is only needed for local deployment
10
  gradio
 
6
  diffusers
7
  transformers
8
  scipy
9
+ spaces
10
  # This is only needed for local deployment
11
  gradio