Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,7 +70,7 @@ def create_image(image_pil,
|
|
| 70 |
|
| 71 |
style_image = Image.open(image_pil).convert('RGB')
|
| 72 |
|
| 73 |
-
|
| 74 |
image = pipe(
|
| 75 |
width=1024,
|
| 76 |
height=1024,
|
|
@@ -82,10 +82,14 @@ def create_image(image_pil,
|
|
| 82 |
clip_image=style_image,
|
| 83 |
ipadapter_scale=scale,
|
| 84 |
).images[0]
|
| 85 |
-
|
|
|
|
| 86 |
if torch.cuda.is_available():
|
| 87 |
torch.cuda.empty_cache()
|
| 88 |
gc.collect()
|
|
|
|
|
|
|
|
|
|
| 89 |
|
| 90 |
return image
|
| 91 |
|
|
|
|
| 70 |
|
| 71 |
style_image = Image.open(image_pil).convert('RGB')
|
| 72 |
|
| 73 |
+
print("starting inference")
|
| 74 |
image = pipe(
|
| 75 |
width=1024,
|
| 76 |
height=1024,
|
|
|
|
| 82 |
clip_image=style_image,
|
| 83 |
ipadapter_scale=scale,
|
| 84 |
).images[0]
|
| 85 |
+
|
| 86 |
+
print("collecting")
|
| 87 |
if torch.cuda.is_available():
|
| 88 |
torch.cuda.empty_cache()
|
| 89 |
gc.collect()
|
| 90 |
+
|
| 91 |
+
print("printing image")
|
| 92 |
+
print(image)
|
| 93 |
|
| 94 |
return image
|
| 95 |
|