Spaces:
Runtime error
Runtime error
bankholdup
commited on
Commit
•
b0d30c3
1
Parent(s):
f144f5c
Update app.py
Browse files
app.py
CHANGED
@@ -123,24 +123,30 @@ def inference(img, model):
|
|
123 |
random_seed = round(time.time() * 1000)
|
124 |
set_seed(random_seed)
|
125 |
|
126 |
-
|
127 |
-
|
128 |
-
input_image = run_alignment('out.jpg')
|
129 |
-
transformed_image = transform(input_image)
|
130 |
-
|
131 |
-
ffhq_codes = ffhq_encoder(transformed_image.unsqueeze(0).to(device).float())
|
132 |
-
ffhq_codes = ffhq_codes + ffhq_latent_avg.repeat(ffhq_codes.shape[0], 1, 1)
|
133 |
-
|
134 |
-
cat_codes = cat_encoder(transformed_image.unsqueeze(0).to(device).float())
|
135 |
-
cat_codes = cat_codes + cat_latent_avg.repeat(cat_codes.shape[0], 1, 1)
|
136 |
-
|
137 |
-
dog_codes = dog_encoder(transformed_image.unsqueeze(0).to(device).float())
|
138 |
-
dog_codes = dog_codes + dog_latent_avg.repeat(dog_codes.shape[0], 1, 1)
|
139 |
|
140 |
-
|
141 |
-
|
142 |
-
|
143 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
144 |
|
145 |
title = "PetBreeder v1.1"
|
146 |
description = "Gradio Demo for PetBreeder. Based on [Colab](https://colab.research.google.com/github/tg-bomze/collection-of-notebooks/blob/master/PetBreeder.ipynb) by [@MLArt](https://t.me/MLArt)."
|
|
|
123 |
random_seed = round(time.time() * 1000)
|
124 |
set_seed(random_seed)
|
125 |
|
126 |
+
try:
|
127 |
+
img.save('out.jpg')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
128 |
|
129 |
+
try:
|
130 |
+
input_image = run_alignment('out.jpg')
|
131 |
+
except:
|
132 |
+
return 'out.jpg'
|
133 |
+
transformed_image = transform(input_image)
|
134 |
+
|
135 |
+
ffhq_codes = ffhq_encoder(transformed_image.unsqueeze(0).to(device).float())
|
136 |
+
ffhq_codes = ffhq_codes + ffhq_latent_avg.repeat(ffhq_codes.shape[0], 1, 1)
|
137 |
+
|
138 |
+
cat_codes = cat_encoder(transformed_image.unsqueeze(0).to(device).float())
|
139 |
+
cat_codes = cat_codes + cat_latent_avg.repeat(cat_codes.shape[0], 1, 1)
|
140 |
+
|
141 |
+
dog_codes = dog_encoder(transformed_image.unsqueeze(0).to(device).float())
|
142 |
+
dog_codes = dog_codes + dog_latent_avg.repeat(dog_codes.shape[0], 1, 1)
|
143 |
+
|
144 |
+
npimage = gen_im(ffhq_codes, dog_codes, cat_codes, model)
|
145 |
+
|
146 |
+
imageio.imwrite('filename.jpeg', npimage)
|
147 |
+
return 'filename.jpeg'
|
148 |
+
except:
|
149 |
+
pass
|
150 |
|
151 |
title = "PetBreeder v1.1"
|
152 |
description = "Gradio Demo for PetBreeder. Based on [Colab](https://colab.research.google.com/github/tg-bomze/collection-of-notebooks/blob/master/PetBreeder.ipynb) by [@MLArt](https://t.me/MLArt)."
|