Update main_dl.py

main_dl.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, Header, Depends, Request
+from fastapi import FastAPI, HTTPException, Header, Depends, Request, Response
 from fastapi.responses import JSONResponse
 from fastapi.security import HTTPBasic, HTTPBasicCredentials
 from fastapi.exceptions import RequestValidationError
@@ -310,7 +310,12 @@ def check_api(lang_tgt:str,
         else:
             translation_model = transformer_en_fr
     plot_model(translation_model, show_shapes=True, show_layer_names=True, show_layer_activations=True,rankdir='TB',to_file=imagePath+'/model_plot.png')
-    return 
+    with open(imagePath+'/model_plot.png', "rb") as image_file:
+            # Lire les données de l'image
+            image_data = image_file.read()
+            
+            # Retourner l'image en tant que réponse HTTP avec le type de contenu approprié
+            return Response(content=image_data, media_type="image/png")
 ''' 
 def run():