update app

app.py

@@ -1,19 +1,18 @@
 import gradio as gr
 import os
-from utils import extraire_informations_carte
-
-def predict(img, type_doc):
-    data = {"Nouvelle_CNI":1,"ANCIENNE_CNI":2,"PERMIS_DE_CONDUITE":3}
-    type_document = data[type_doc]
-    result = extraire_informations_carte(img,type_document)
-    return result
-
-
-
+from utils import extraire_informations_carte,make_prediction
 
+def predict(img):
+    proba, document_type = make_prediction(img)
+    if proba < .98:
+        return "Ce document ne fait pas partir de ceux pris en charge actuelement (Nouvelle et Ancienne CNI, Permis de conduire)"
+    else :
+        doc_type = document_type +1
+        result = extraire_informations_carte(img,doc_type)
+        return result
 
 image = gr.components.Image(type = "filepath")
-type_document = gr.components.Dropdown(["Nouvelle_CNI","ANCIENNE_CNI","PERMIS_DE_CONDUITE"])
+#type_document = gr.components.Dropdown(["Nouvelle_CNI","ANCIENNE_CNI","PERMIS_DE_CONDUITE"])
 out_lab = gr.components.Textbox()
 
 ### 4. Gradio app ###
@@ -23,15 +22,15 @@ description = "WE USE OCR TO EXTRACT INFORMATIONS FROM DIFFERENT TYPES OF DOCUME
 article = "Created by data354."
 
 # Create examples list from "examples/" directory
-example_list = [["examples/" + example,i] for example,i in zip(os.listdir("examples"),["ANCIENNE_CNI","ANCIENNE_CNI","Nouvelle_CNI","Nouvelle_CNI","Nouvelle_CNI","PERMIS_DE_CONDUITE","PERMIS_DE_CONDUITE"])]
+example_list = [["examples/" + example] for example in os.listdir("examples")]
 print(example_list)
 #[gr.Label(label="Predictions"), # what are the outputs?
 #gr.Number(label="Prediction time (s)")], # our fn has two outputs, therefore we have two outputs
 # Create examples list from "examples/" directory
 # Create the Gradio demo
 demo = gr.Interface(fn=predict, # mapping function from input to output
-                    inputs= [image,type_document], #gr.Image(type="pil"), # what are the inputs?
-                    outputs=out_lab,
+                    inputs= image, #gr.Image(type="pil"), # what are the inputs?
+                    outputs=out_lab, #[list of outputs]
                     examples=example_list, 
                     title=title,
                     description=description,