tweaks

app.py

@@ -10,7 +10,6 @@ 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'
@@ -114,9 +113,8 @@ 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}:{url_shortener}')
+                axs[idx, j].set_title(f'{y}{j}:{x.url}')
                 axs[idx, j].axis('off')
 
                 x_im = preprocess(x_im) # (3, H, W)

requirements.txt

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