debisoft commited on
Commit
129ae32
·
1 Parent(s): 8496ec0
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,6 +13,7 @@ from matplotlib.animation import FuncAnimation, PillowWriter
13
  import numpy as np
14
  from IPython.display import HTML
15
  from diffusion_utilities import *
 
16
 
17
  #openai.api_key = os.getenv('OPENAI_API_KEY')
18
 
@@ -162,7 +163,7 @@ def sample_ddim(n_sample, n=20):
162
 
163
  def greet(input):
164
  samples, intermediate = sample_ddim(32, n=25)
165
- response = intermediate[-1][-1][-1];
166
  return response
167
 
168
  #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
13
  import numpy as np
14
  from IPython.display import HTML
15
  from diffusion_utilities import *
16
+ from PIL import Image as im
17
 
18
  #openai.api_key = os.getenv('OPENAI_API_KEY')
19
 
 
163
 
164
  def greet(input):
165
  samples, intermediate = sample_ddim(32, n=25)
166
+ response = im.fromarray(intermediate[-1][-1][-1]);
167
  return response
168
 
169
  #iface = gr.Interface(fn=greet, inputs="text", outputs="text")