Commit
·
11d974d
1
Parent(s):
c03d5bb
Update app.py
Browse files
app.py
CHANGED
@@ -187,6 +187,7 @@ def inference(
|
|
187 |
print(f"Inference ended at {end_time_formatted}, taking {end_time-start_time}s")
|
188 |
logger.debug("Output Types: generated_image=%s", type(None))
|
189 |
logger.debug("Output Values: generated_image=None")
|
|
|
190 |
return tensor_to_pil_image(out_image["images"][0])
|
191 |
|
192 |
|
|
|
187 |
print(f"Inference ended at {end_time_formatted}, taking {end_time-start_time}s")
|
188 |
logger.debug("Output Types: generated_image=%s", type(None))
|
189 |
logger.debug("Output Values: generated_image=None")
|
190 |
+
print(out_image["images"][0].shape)
|
191 |
return tensor_to_pil_image(out_image["images"][0])
|
192 |
|
193 |
|