feat: Support navigate from HuggingFace

j3soon · Aug 4, 2024 · 8096f62 · 8096f62
1 parent b4e0f76
commit 8096f62
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/chrome/target_url_regexp_replace.js b/chrome/target_url_regexp_replace.js
@@ -12,6 +12,7 @@ export default [
   [/^.*:\/\/(?:export\.|browse\.|www\.)?arxiv\.org\/ftp\/(?:arxiv\/|([^\/]*\/))papers\/.*?([^\/]*?)\.pdf(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1$2"],
   [/^.*:\/\/(?:browse\.|www\.)?arxiv\.org\/html\/(\S*?)\/*(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1"],
   [/^.*:\/\/ar5iv\.labs\.arxiv\.org\/html\/(\S*?)\/*(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1"],
+  [/^.*:\/\/huggingface\.co\/papers\/(\S*?)(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1"],
   [/^.*:\/\/openreview\.net\/forum\?(?:.*?&)?id=(\S*?)(&.*?)?(\#.*?)?$/, "https://openreview.net/pdf?id=$1"],
   [/^.*:\/\/openreview\.net\/pdf\?(?:.*?&)?id=(\S*?)(&.*?)?(\#.*?)?$/, "https://openreview.net/forum?id=$1"],
   // Starting from 2022, NIPS urls may end with a `-Conference` suffix

diff --git a/firefox/target_url_regexp_replace.js b/firefox/target_url_regexp_replace.js
@@ -12,6 +12,7 @@ export default [
   [/^.*:\/\/(?:export\.|browse\.|www\.)?arxiv\.org\/ftp\/(?:arxiv\/|([^\/]*\/))papers\/.*?([^\/]*?)\.pdf(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1$2"],
   [/^.*:\/\/(?:browse\.|www\.)?arxiv\.org\/html\/(\S*?)\/*(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1"],
   [/^.*:\/\/ar5iv\.labs\.arxiv\.org\/html\/(\S*?)\/*(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1"],
+  [/^.*:\/\/huggingface\.co\/papers\/(\S*?)(\?.*?)?(\#.*?)?$/, "https://arxiv.org/abs/$1"],
   [/^.*:\/\/openreview\.net\/forum\?(?:.*?&)?id=(\S*?)(&.*?)?(\#.*?)?$/, "https://openreview.net/pdf?id=$1"],
   [/^.*:\/\/openreview\.net\/pdf\?(?:.*?&)?id=(\S*?)(&.*?)?(\#.*?)?$/, "https://openreview.net/forum?id=$1"],
   // Starting from 2022, NIPS urls may end with a `-Conference` suffix

diff --git a/tests/testcases/testcases.yaml b/tests/testcases/testcases.yaml
@@ -175,6 +175,11 @@ navigation:
   url2: https://arxiv.org/abs/1512.03385
   title2: Deep Residual Learning for Image Recognition | Abstract
   description: ar5iv -> arXiv
+- url: https://huggingface.co/papers/2408.00714
+  title: "Paper page - SAM 2: Segment Anything in Images and Videos"
+  url2: https://arxiv.org/abs/2408.00714
+  title2: "SAM 2: Segment Anything in Images and Videos | Abstract"
+  description: ar5iv -> arXiv
 - url: https://openreview.net/pdf?id=LcF-EEt8cCC
   url2: https://openreview.net/forum?id=LcF-EEt8cCC
   title2: Denoising Likelihood Score Matching for Conditional Score-based Data Generation | OpenReview
@@ -321,11 +326,10 @@ navigation:
   pdf_url: https://ieeexplore.ieee.org/stamp/stamp.jsp?arnumber=10412086
   skip_selenium: True # skipped due to paywall
   description: IEEE
-- url: https://ieeexplore.ieee.org/document/10412086
-  pdf_url: https://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=10412086&tag=1
-  abs2pdf: False
+- url: https://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=10412086&tag=1
+  url2: https://ieeexplore.ieee.org/document/10412086
   skip_selenium: True # skipped due to paywall
-  description: IEEE (with extra query strings)
+  description: IEEE pdf -> abs (with extra query strings)
 
 # TODO: Test download filenames:
 #       - Title with colon: https://arxiv.org/abs/2102.07936