Update app.py

app.py

@@ -90,17 +90,17 @@ def predict_dnn(input):
     print(f"Prediction: {prediction}")
     return prediction
 
-def main(model):
+def main(model, audio):
     if model == "DNN":
-        return predict_dnn(input)
+        return predict_dnn(audio)
     elif model == "RandomForest":
-        return predict_rf(input)
+        return predict_rf(audio)
 
 
-with gr.Blocks() as demo:
-    model = gr.Dropdown(choices=model_names),
-    main,
+iface = gr.Interface(
+    fn = main,
     inputs=[
+        gr.Dropdown(choices=model_names),
         gr.Audio(sources=["microphone","upload"], type="filepath")
     ],
     outputs=[
@@ -108,6 +108,6 @@ with gr.Blocks() as demo:
     ],
     live=True,
     description="demo for Audio to question classifier"
-    
+)
 
-demo.launch()
+iface.launch()