Fix #19 app.py

app.py

@@ -8,7 +8,7 @@ import os
 from types import SimpleNamespace
 import random
 from torchvision.utils import save_image
-import gradio as gr  # Importamos Gradio
+import gradio as gr
 
 # Asegúrate de que las funciones necesarias estén definidas (si no lo están ya)
 def resize(img, size):
@@ -274,12 +274,15 @@ class Solver(object):
             # Generar la imagen con el estilo transferido
             generated_image = self.G(source_image, s_ref)
 
-            # Denormalizar la imagen para mostrarla o guardarla
+            # Denormalizar la imagen para mostrarla en la interfaz
             generated_image = denormalize(generated_image.squeeze(0)).cpu()
             return generated_image
 
 # Función principal para la inferencia
-def main(args, checkpoint_path, source_image, reference_image, target_domain_index): # Cambiamos los paths por las imágenes
+def main(source_image, reference_image, target_domain_index, checkpoint_path, args):
+    if source_image is None or reference_image is None:
+        raise gr.Error("Por favor, proporciona ambas imágenes (fuente y referencia).")
+
     # Crear el solver
     solver = Solver(args)
     # Cargar el checkpoint
@@ -287,7 +290,6 @@ def main(args, checkpoint_path, source_image, reference_image, target_domain_ind
 
     # Realizar la transferencia de estilo
     generated_image = solver.transfer_style(source_image, reference_image, target_domain_index)
-
     return generated_image
 
 def gradio_interface():
@@ -327,4 +329,4 @@ def gradio_interface():
 
 if __name__ == '__main__':
     iface = gradio_interface()
-    iface.launch()
+    iface.launch(share=True)