Prabin1 commited on
Commit
5448ba6
·
verified ·
1 Parent(s): 38f41fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -20
app.py CHANGED
@@ -6,31 +6,28 @@ from model_utils import load_model, prepare_input_for_model, interpret_predictio
6
  MODEL_PATH = "best_audio_model.keras"
7
 
8
  # Load model once when the app starts
9
-
10
  model = load_model(MODEL_PATH)
11
 
12
  def predict_audio(file_path):
13
- try:
14
- features = extract_features_with_time_series(file_path)
15
- if features is None:
16
- return {"Error": "Feature extraction failed. Please check the audio."}
17
 
18
- ```
19
- input_data = prepare_input_for_model(features, model)
20
- raw_pred = model.predict(input_data)
21
- result = interpret_prediction(raw_pred)
22
- return result
23
- except Exception as e:
24
- return {"Error": str(e)}
25
- ```
26
 
27
  iface = gr.Interface(
28
- fn=predict_audio,
29
- inputs=gr.Audio(type="filepath", label="Upload an audio file"),
30
- outputs=gr.Label(num_top_classes=2, label="Prediction"),
31
- title="Audio Deepfake Detection",
32
- description="Upload an audio clip to detect whether it is Real or Fake using a Keras model."
33
  )
34
 
35
- if **name** == "**main**":
36
- iface.launch(server_name="0.0.0.0", server_port=7860)
 
6
  MODEL_PATH = "best_audio_model.keras"
7
 
8
  # Load model once when the app starts
 
9
  model = load_model(MODEL_PATH)
10
 
11
  def predict_audio(file_path):
12
+ try:
13
+ features = extract_features_with_time_series(file_path)
14
+ if features is None:
15
+ return {"Error": "Feature extraction failed. Please check the audio."}
16
 
17
+ input_data = prepare_input_for_model(features, model)
18
+ raw_pred = model.predict(input_data)
19
+ result = interpret_prediction(raw_pred)
20
+ return result
21
+ except Exception as e:
22
+ return {"Error": str(e)}
 
 
23
 
24
  iface = gr.Interface(
25
+ fn=predict_audio,
26
+ inputs=gr.Audio(type="filepath", label="Upload an audio file"),
27
+ outputs=gr.Label(num_top_classes=2, label="Prediction"),
28
+ title="Audio Deepfake Detection",
29
+ description="Upload an audio clip to detect whether it is Real or Fake using a Keras model."
30
  )
31
 
32
+ if __name__ == "__main__":
33
+ iface.launch(server_name="0.0.0.0", server_port=7860