Update app.py

app.py

@@ -7,17 +7,21 @@ from markdownify import markdownify
 
 models = {
     "jinaai/reader-lm-0.5b": AutoModelForCausalLM.from_pretrained("jinaai/reader-lm-0.5b", trust_remote_code=True).eval().to("cuda"),
-    "jinaai/reader-lm-1.5b": AutoModelForCausalLM.from_pretrained("jinaai/reader-lm-1.5b", trust_remote_code=True).eval().to("cuda")
+    "jinaai/reader-lm-1.5b": AutoModelForCausalLM.from_pretrained("jinaai/reader-lm-1.5b", trust_remote_code=True).eval().to("cuda"),
+    "jinaai/ReaderLM-v2": AutoModelForCausalLM.from_pretrained("jinaai/ReaderLM-v2", trust_remote_code=True).eval().to("cuda")
+
 }
 
 tokenizers = {
     "jinaai/reader-lm-0.5b": AutoTokenizer.from_pretrained("jinaai/reader-lm-0.5b", trust_remote_code=True),
     "jinaai/reader-lm-1.5b": AutoTokenizer.from_pretrained("jinaai/reader-lm-1.5b", trust_remote_code=True),
+    "jinaai/ReaderLM-v2": AutoTokenizer.from_pretrained("jinaai/ReaderLM-v2", trust_remote_code=True),
+
 }
 
 
 @spaces.GPU
-def run_example(html_content, model_id="jinaai/reader-lm-1.5b"):
+def run_example(html_content, model_id="jinaai/ReaderLM-v2"):
     print("Start Model Processing")
     model = models[model_id]
     tokenizer = tokenizers[model_id]