Update app.py

app.py

@@ -92,10 +92,11 @@ def predict_dnn(input):
 
 def main(model, audio):
     if model == "DNN":
-        return predict_dnn(audio)
+        predict = predict_dnn(audio)
+        
     elif model == "RandomForest":
-        return predict_rf(audio)
-
+        predict = predict_rf(audio)
+    return predict
 
 iface = gr.Interface(
     fn = main,