fixing broken predictions - debug prints enabled

app.py

@@ -35,4 +35,4 @@ demo = gr.Interface(
     ],
     description=description
 )
-demo.launch(share=True,debug=True)
+demo.launch(debug=True)

predict.py

@@ -30,8 +30,10 @@ def predict_image(image, processor, model):
   prediction = model.config.id2label[predicted_class_idx]
 
   if prediction.lower() == "real":
+    print("real ")
     return {"real": 1.0, "fake": 0.0}  # 100% confidence for real
   else:  # prediction == "fake"
+    print("fake ")
     return {"real": 0.0, "fake": 1.0}
 
 

process_audio.py

@@ -91,6 +91,7 @@ def create_mel_spectrograms(file_path, segment_duration, start_offset):
             S_dB = librosa.power_to_db(S, ref=np.max)
 
             # Setup axis
+            img = librosa.display.specshow(S_dB, sr=sr, fmax=8000, ax=ax)
             ax.set_xlim(0, S.shape[-1])
             ax.set_ylim(0, S.shape[0])
             ax.set_xticks([])