Update README.md

README.md

@@ -31,12 +31,12 @@ model.to(device)
 
 context_length = 2048
 
-def predict(image):
+def predict(img_path):
     # prepare PDF image for the model
-    image = Image.open(image)
+    image = Image.open(img_path)
     pixel_values = processor(image, return_tensors="pt").pixel_values
 
-    # generate transcription (here we only generate 30 tokens)
+    # generate transcription
     outputs = model.generate(
         pixel_values.to(device),
         min_length=1,