Christ20 commited on
Commit
05fa210
·
verified ·
1 Parent(s): c201696

Upload 3 files

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. Model_1.keras +3 -0
  3. app.py +34 -0
  4. requirements.txt +0 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ Model_1.keras filter=lfs diff=lfs merge=lfs -text
Model_1.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db61448c8203d7db0e7cd6e173626caa23e8958b77b2243d8c87f98b62a2ddab
3
+ size 10807626
app.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from tensorflow.keras.models import load_model
3
+ from tensorflow.keras.preprocessing import image
4
+ import numpy as np
5
+ from PIL import Image
6
+
7
+
8
+ model = load_model("Model_1.keras")
9
+
10
+
11
+ def predict_image(img):
12
+ img = img.resize((256, 256))
13
+ img_array = np.array(img) / 255.0
14
+ img_array = np.expand_dims(img_array, axis=0)
15
+ prediction = model.predict(img_array)
16
+ class_names = ["Fake","Real"]
17
+ predicted_class = np.argmax(prediction[0])
18
+ confidence = prediction[0][predicted_class]
19
+ if confidence < 0.49:
20
+ return f"C'est une image générée par un Algorithme d'IA, avec une confiance de {confidence*100: .2f}% "
21
+ elif confidence > 0.52:
22
+ return f"C'est une image réelle, avec une confiance de {confidence*100:.2f}% "
23
+ else:
24
+ return f"Plus ou moins une image rélle, avec une confiance de {confidence*100: .2f}% "
25
+
26
+ interface = gr.Interface(
27
+ fn=predict_image,
28
+ inputs=gr.Image(type="pil"),
29
+ outputs="text",
30
+ title="Détection d'images générée par l'IA'"
31
+ )
32
+
33
+
34
+ interface.launch("share=True")
requirements.txt ADDED
Binary file (2.96 kB). View file