Update app.py

app.py

@@ -16,12 +16,7 @@ def predict_image(img):
     class_names = ["Fake","Real"]
     predicted_class = np.argmax(prediction[0])
     confidence = prediction[0][predicted_class]
-    if confidence < 0.49:
-        return f"C'est une image générée par un Algorithme d'IA, avec une confiance de {confidence*100: .2f}% "
-    elif confidence > 0.52:
-        return f"C'est une image réelle, avec une confiance de {confidence*100:.2f}%  "
-    else:
-        return f"Plus ou moins une image rélle, avec une confiance de {confidence*100: .2f}%  "
+    return f"Prediction: {class_names[predicted_class]} (Confidence: {confidence:.2f})"
 
 interface = gr.Interface(
     fn=predict_image,