Update app.py

app.py

@@ -11,7 +11,7 @@ import gradio as gr
 import torchaudio
 
 # Construct pairs of text and image
-training_pairs = make_pairs(CFG.train_annotation_file, CFG.image_dir, 5) # 413.915 -> 82.783 images
+training_pairs = make_pairs(CFG.image_dir, CFG.image_dir, 5) # 413.915 -> 82.783 images
 
 # Sorted according images
 training_pairs = sorted(training_pairs, key=lambda x: x[0])