Update app.py

app.py

@@ -57,6 +57,11 @@ model = CLIPModel().to(device)
 model.load_state_dict(torch.load("weights.pt", map_location=torch.device('cpu')))
 text_embeddings = torch.load('saved_text_embeddings.pt', map_location=device)
 
+def download_link(content, filename, link_text):
+    b64 = base64.b64encode(content).decode()
+    href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{link_text}</a>'
+    return href
+
 def show_predicted_caption(image, top_k=1):
     matches = predict_caption(
         image, model, text_embeddings, testing_df["caption"]