Update app.py
Browse files
app.py
CHANGED
|
@@ -2,9 +2,6 @@ import gradio as gr
|
|
| 2 |
from image_backend import predict_image_pil
|
| 3 |
from audio_backend import predict_audio
|
| 4 |
|
| 5 |
-
# Add Material Icons stylesheet
|
| 6 |
-
gr.HTML('<link href="https://fonts.googleapis.com/icon?family=Material+Icons" rel="stylesheet">')
|
| 7 |
-
|
| 8 |
# =========================
|
| 9 |
# IMAGE LOGIC
|
| 10 |
# =========================
|
|
@@ -50,16 +47,19 @@ def analyze_audio(audio_path):
|
|
| 50 |
# =========================
|
| 51 |
# UI
|
| 52 |
# =========================
|
| 53 |
-
with gr.Blocks() as demo:
|
| 54 |
-
|
|
|
|
|
|
|
|
|
|
| 55 |
|
| 56 |
with gr.Tabs():
|
| 57 |
# HOME TAB
|
| 58 |
-
with gr.Tab('<
|
| 59 |
-
gr.Markdown("## Welcome
|
| 60 |
|
| 61 |
# IMAGE TAB
|
| 62 |
-
with gr.Tab('<
|
| 63 |
gr.Markdown("## Deepfake Image Detection System")
|
| 64 |
with gr.Row():
|
| 65 |
with gr.Column(scale=1):
|
|
@@ -77,7 +77,7 @@ with gr.Blocks() as demo:
|
|
| 77 |
outputs=[image_input, img_pred, img_conf, img_risk])
|
| 78 |
|
| 79 |
# AUDIO TAB
|
| 80 |
-
with gr.Tab('<
|
| 81 |
gr.Markdown("## Deepfake Audio Detection System")
|
| 82 |
with gr.Row():
|
| 83 |
with gr.Column(scale=1):
|
|
|
|
| 2 |
from image_backend import predict_image_pil
|
| 3 |
from audio_backend import predict_audio
|
| 4 |
|
|
|
|
|
|
|
|
|
|
| 5 |
# =========================
|
| 6 |
# IMAGE LOGIC
|
| 7 |
# =========================
|
|
|
|
| 47 |
# =========================
|
| 48 |
# UI
|
| 49 |
# =========================
|
| 50 |
+
with gr.Blocks(css="style.css") as demo:
|
| 51 |
+
# Load Material Icons stylesheet
|
| 52 |
+
gr.HTML('<link href="https://fonts.googleapis.com/icon?family=Material+Icons" rel="stylesheet">')
|
| 53 |
+
|
| 54 |
+
gr.Markdown("# AI Driven Deepfake Detection System")
|
| 55 |
|
| 56 |
with gr.Tabs():
|
| 57 |
# HOME TAB
|
| 58 |
+
with gr.Tab('<span class="material-icons">home</span> Home'):
|
| 59 |
+
gr.Markdown("## Welcome\nSelect the type of media you want to analyze:")
|
| 60 |
|
| 61 |
# IMAGE TAB
|
| 62 |
+
with gr.Tab('<span class="material-icons">image</span> Image Detection'):
|
| 63 |
gr.Markdown("## Deepfake Image Detection System")
|
| 64 |
with gr.Row():
|
| 65 |
with gr.Column(scale=1):
|
|
|
|
| 77 |
outputs=[image_input, img_pred, img_conf, img_risk])
|
| 78 |
|
| 79 |
# AUDIO TAB
|
| 80 |
+
with gr.Tab('<span class="material-icons">audiotrack</span> Audio Detection'):
|
| 81 |
gr.Markdown("## Deepfake Audio Detection System")
|
| 82 |
with gr.Row():
|
| 83 |
with gr.Column(scale=1):
|