Fix input handling and update interface for multiple models

app.py

@@ -6,13 +6,13 @@ audio_model = pipeline("audio-classification", model="MelodyMachine/Deepfake-aud
 image_model = pipeline("image-classification", model="dima806/deepfake_vs_real_image_detection")
 
 # Define the prediction function
-def predict(data, model_choice):
-    print("Data received:", data)  # Debugging statement
+def predict(audio, image, model_choice):
+    print("Data received:", audio if model_choice == "Audio Deepfake Detection" else image)  # Debugging statement
     try:
         if model_choice == "Audio Deepfake Detection":
-            result = audio_model(data)
+            result = audio_model(audio)
         elif model_choice == "Image Deepfake Detection":
-            result = image_model(data)
+            result = image_model(image)
         else:
             return {"error": "Invalid model choice"}
         
@@ -42,6 +42,6 @@ with gr.Blocks() as iface:
     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=[gr.State(audio_input), gr.State(image_input), model_choice], outputs=output)
+    submit_button.click(fn=predict, inputs=[audio_input, image_input, model_choice], outputs=output)
     
     iface.launch()