minor update

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from PIL import Image, ImageOps
 import numpy as np
+from tibblingai import wta
 import matplotlib.pyplot as plt  # needed for plt.imread
 def load_sample_images():
     sample_paths = ["sample1.jpg", "sample2.jpg"]  # Must be in the same folder as your script