img2img response

api/core/controllers/text2image.py

@@ -3,23 +3,43 @@ import torch
 
 class Generador:
     def using_runway_sd_15(prompt:str)->list:
-        _generador = Pipe.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
-        _generador.to("cuda")
-        _imagen = _generador(prompt).images[0]
-        return list(_imagen.getdata())
+        try:
+            _generador = Pipe.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+            _generador.to("cuda")
+            _imagen = _generador(prompt).images[0]
+            _response = list(_imagen.getdata())
+        except Exception as e:
+            _response = list(str(e))
+        finally:
+            return _response
     def using_stability_sd_21(prompt:str)->list:
-        _generador = Pipe.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
-        _generador.to("cuda")
-        _imagen = _generador(prompt).images[0]
-        return list(_imagen.getdata())
+        try:
+            _generador = Pipe.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
+            _generador.to("cuda")
+            _imagen = _generador(prompt).images[0]
+            _response = list(_imagen.getdata())
+        except Exception as e:
+            _response = list(str(e))
+        finally:
+            return _response
     def using_realistic_v14(prompt:str)->list:
-        _generador = Pipe.from_pretrained("SG161222/Realistic_Vision_V1.4", torch_dtype=torch.float16)
-        _generador.to("cuda")
-        _imagen = _generador(prompt).images[0]
-        return list(_imagen.getdata())
+        try:
+            _generador = Pipe.from_pretrained("SG161222/Realistic_Vision_V1.4", torch_dtype=torch.float16)
+            _generador.to("cuda")
+            _imagen = _generador(prompt).images[0]
+            _response = list(_imagen.getdata())
+        except Exception as e:
+            _response = list(str(e))
+        finally:
+            return _response
     def using_prompthero_openjourney(prompt:str)->list:
-        _generador = Pipe.from_pretrained("prompthero/openjourney", torch_dtype=torch.float16)
-        _generador.to("cuda")
-        _imagen = _generador(prompt).images[0]
-        return list(_imagen.getdata())
+        try:
+            _generador = Pipe.from_pretrained("prompthero/openjourney", torch_dtype=torch.float16)
+            _generador.to("cuda")
+            _imagen = _generador(prompt).images[0]
+            _response = list(_imagen.getdata())
+        except Exception as e:
+            _response = list(str(e))
+        finally:
+            return _response
     

api/main.py

@@ -88,7 +88,8 @@ def get_text2img(data:dict) -> dict:
     try:
         if data:
             __response['original']= data.get('texto')
-            __response['image']= __main.text_to_img(texto=data.get('texto'))
+            __response['image']= __main.text_to_img(texto=data.get('texto'), 
+                                                    model=data.get('modelo'))
         else:
             raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
     except Exception as e: