initial commit

app.py

@@ -13,7 +13,7 @@ model.config.pad_token_id = tokenizer.eos_token_id
 
 @spaces.GPU
 def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, render=False):
-    # Create attention mask
+
     attention_mask = torch.ones((1, model.config.max_position_embeddings), dtype=torch.long, device=model.device)
     
     if task == "Plain Text OCR":