Fix #28 app.py

app.py

@@ -11,7 +11,7 @@ from torchvision.utils import save_image
 import gradio as gr
 import numpy as np
 import io
-import tempfile
+import tempfile  # Importar tempfile
 
 # Asegúrate de que las funciones necesarias estén definidas (si no lo están ya)
 def resize(img, size):
@@ -266,19 +266,9 @@ class Solver(object):
                 transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
             ])
             # Convertir a PIL image antes de la transformación
-            try:
-                with tempfile.NamedTemporaryFile(suffix=".jpg") as source_temp_file, \
-                        tempfile.NamedTemporaryFile(suffix=".jpg") as reference_temp_file:
-                    source_temp_file.write(source_image)
-                    reference_temp_file.write(reference_image)
-                    # Verificar el contenido de los archivos temporales
-                    print(f"Primeros 100 bytes de {source_temp_file.name}: {open(source_temp_file.name, 'rb').read(100)}")
-                    print(f"Primeros 100 bytes de {reference_temp_file.name}: {open(reference_temp_file.name, 'rb').read(100)}")
-                    source_image_pil = Image.open(source_temp_file.name)
-                    reference_image_pil = Image.open(reference_temp_file.name)
-            except Exception as e:
-                print(f"Error al procesar las imágenes: {e}")
-                raise  # Re-raise la excepción para que Gradio la capture
+            source_image_pil = Image.fromarray(source_image)
+            reference_image_pil = Image.fromarray(reference_image)
+
             source_image = transform(source_image_pil).unsqueeze(0).to(self.device)
             reference_image = transform(reference_image_pil).unsqueeze(0).to(self.device)