Spaces:
Sleeping
Sleeping
app.py
CHANGED
@@ -163,7 +163,7 @@ def sample_ddim(n_sample, n=20):
|
|
163 |
|
164 |
def greet(input):
|
165 |
samples, intermediate = sample_ddim(32, n=25)
|
166 |
-
transformImg =
|
167 |
response = transformImg(transform(intermediate[-1][-1]))
|
168 |
return response
|
169 |
|
|
|
163 |
|
164 |
def greet(input):
|
165 |
samples, intermediate = sample_ddim(32, n=25)
|
166 |
+
transformImg = transforms.ToPILImage()
|
167 |
response = transformImg(transform(intermediate[-1][-1]))
|
168 |
return response
|
169 |
|