Update app.py

app.py

@@ -3,13 +3,13 @@ import gradio as gr
 import pdfplumber
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-model_name = "ibm-granite/granite-docling-258m-demo"
+model_name = "ibm-granite/granite-docling-258M"
 
-# Use the Hugging Face token stored in Secrets
+# Use HF token stored in Space secrets
 hf_token = os.environ.get("HF_HUB_TOKEN")
 
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=hf_token)
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name, token=hf_token)
 
 def extract_text_from_pdf(pdf_file):
     text = ""
@@ -32,7 +32,7 @@ interface = gr.Interface(
     inputs=gr.File(file_types=[".pdf"]),
     outputs=gr.Textbox(label="Generated JSON"),
     title="PDF to JSON using Granite DocLing",
-    description="Upload a PDF and get a JSON output using the ibm-granite/granite-docling-258m-demo model."
+    description="Upload a PDF and get a JSON output using ibm-granite/granite-docling-258M."
 )
 
 interface.launch()