show short version of url

app.py

@@ -10,6 +10,7 @@ import matplotlib.pyplot as plt
 from models import get_model
 from dotmap import DotMap
 from PIL import Image
+from pyshorteners import Shortener
 
 #os.environ['TERM'] = 'linux'
 #os.environ['TERMINFO'] = '/etc/terminfo'
@@ -113,8 +114,9 @@ def inference(query, labels, n_supp=10):
                 x_im = Image.open(x.path)
 
                 # vis
+                url_shortener = Shortener('Bitly', bitly_token='YOUR TOKEN')
                 axs[idx, j].imshow(x_im)
-                axs[idx, j].set_title(f'{y}{j}:{x.path}')
+                axs[idx, j].set_title(f'{y}{j}:{url_shortener}')
                 axs[idx, j].axis('off')
 
                 x_im = preprocess(x_im) # (3, H, W)

requirements.txt

@@ -10,3 +10,4 @@ ml-collections
 ftfy
 tensorboard
 Google-Images-Search
+pyshorteners