Update app.py

app.py

@@ -7,11 +7,12 @@ from huggingface_hub import HfApi
 import os
 
 PATH_MODEL = os.environ.get('MODEL_REPO_ID')
-
+token = os.environ.get("HF_TOKEN")
+task_prompt = os.environ.get("taskprompt")
 
 # PATH_MODEL = "fruk19/donut_nfact_v4"
-processor = DonutProcessor.from_pretrained(PATH_MODEL)
-model = VisionEncoderDecoderModel.from_pretrained(PATH_MODEL)
+processor = DonutProcessor.from_pretrained(PATH_MODEL, token=token)
+model = VisionEncoderDecoderModel.from_pretrained(PATH_MODEL, token=token)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.eval()
 model.to(device)
@@ -20,7 +21,6 @@ def predict(test_image):
   pixel_values = processor(test_image, return_tensors="pt").pixel_values
   pixel_values = pixel_values.to(device)
 
-  task_prompt = "<s_nfact>"
   decoder_input_ids = processor.tokenizer(task_prompt, add_special_tokens=False, return_tensors="pt").input_ids
   decoder_input_ids = decoder_input_ids.to(device)