Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from transformers import pipeline
 
 # Load the models using pipeline
-audio_model = pipeline("audio-classification", model="MelodyMachine/Deepfake-audio-detection-V2")
 image_model = pipeline("image-classification", model="dima806/deepfake_vs_real_image_detection")
 
 # Define the prediction function
@@ -34,13 +33,9 @@ def update_interface(model_choice):
 
 # Create Gradio interface
 with gr.Blocks() as iface:
- model_choice = gr.Radio(choices=["Audio Deepfake Detection", "Image Deepfake Detection"], label="Select Model", value="Audio Deepfake Detection")
- audio_input = gr.Audio(type="filepath", label="Upload Audio File")
  image_input = gr.Image(type="filepath", label="Upload Image File", visible=False)
  output = gr.Label()
 
- model_choice.change(fn=update_interface, inputs=model_choice, outputs=[audio_input, image_input])
-
  submit_button = gr.Button("Submit")
  submit_button.click(fn=predict, inputs=[audio_input, image_input, model_choice], outputs=output)