atharvaballa commited on
Commit
367b49c
·
verified ·
1 Parent(s): f8249f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
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
- gr.Markdown("# Unified Deepfake Detection System")
 
 
 
55
 
56
  with gr.Tabs():
57
  # HOME TAB
58
- with gr.Tab('<i class="material-icons">home</i> Home'):
59
- gr.Markdown("## Welcome 👋\nSelect the type of media you want to analyze:")
60
 
61
  # IMAGE TAB
62
- with gr.Tab('<i class="material-icons">image</i> Image Detection'):
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('<i class="material-icons">audiotrack</i> Audio Detection'):
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):