tincri commited on
Commit
5eb37b3
1 Parent(s): 12deb67

Fix #30 app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -280,7 +280,7 @@ class Solver(object):
280
  generated_image = self.G(source_image, s_ref)
281
 
282
  # Denormalizar la imagen para mostrarla en la interfaz
283
- generated_image = denormalize(generated_image.squeeze(0)).cpu()
284
  return generated_image
285
 
286
  # Funci贸n principal para la inferencia
 
280
  generated_image = self.G(source_image, s_ref)
281
 
282
  # Denormalizar la imagen para mostrarla en la interfaz
283
+ generated_image = denormalize(generated_image.squeeze(0)).cpu().numpy()
284
  return generated_image
285
 
286
  # Funci贸n principal para la inferencia