Update tasks/audio.py

tasks/audio.py

@@ -72,6 +72,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
     MODEL_PATH = './model'  
     model = tf.keras.models.load_model(MODEL_PATH)
+    model.summary()
     true_labels = test_dataset["label"]
     predictions = []
 
@@ -81,7 +82,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
         pred_probs = model.predict(spectrogram, verbose=0)
         predicted_label = np.argmax(pred_probs)
         predictions.append(predicted_label)
-        
+    print("prediction:", predictions)     
     # Make random predictions (placeholder for actual model inference)
     #true_labels = test_dataset["label"]
     #predictions = [random.randint(0, 1) for _ in range(len(true_labels))]