Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import random
-
 import gradio as gr
 
 #links
@@ -14,6 +13,15 @@ Article = Article + " Originals go here: https://github.com/AaronCWacker/Yggdras
 
 
 def fake_gan():
+    base_dir = os.getcwd()  # Get the current base directory
+    png_files = [file for file in os.listdir(base_dir) if file.endswith(".png")]  # List all files ending with ".png"
+    
+    images = [(random.choice(png_files), f"label {i}" if i != 0 else "label" * 50) for i in range(10)]
+    
+    return images
+
+
+def fake_gan_old():
     images = [
         (random.choice(
             [