Ariamehr commited on
Commit
014db36
1 Parent(s): bf9297c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ from matplotlib.patches import Patch
10
  from PIL import Image, ImageOps
11
  from torchvision import transforms
12
 
13
- !pip install diffusers
14
 
15
  import spaces
16
  from diffusers import DiffusionPipeline
@@ -30,7 +30,7 @@ gr.Interface(
30
 
31
  @spaces.GPU(duration=120)
32
  def generate(prompt):
33
- return pipe(prompt).images
34
 
35
  # ----------------- HELPER FUNCTIONS ----------------- #
36
  os.chdir(os.path.dirname(os.path.abspath(__file__)))
 
10
  from PIL import Image, ImageOps
11
  from torchvision import transforms
12
 
13
+ '''!pip install diffusers
14
 
15
  import spaces
16
  from diffusers import DiffusionPipeline
 
30
 
31
  @spaces.GPU(duration=120)
32
  def generate(prompt):
33
+ return pipe(prompt).images'''
34
 
35
  # ----------------- HELPER FUNCTIONS ----------------- #
36
  os.chdir(os.path.dirname(os.path.abspath(__file__)))