Update requirements1.txt

requirements1.txt

@@ -283,4 +283,61 @@ class Florence2:
                 fp.write(results[task])
                 fp.write("\n")
                 fp.write(f"Image Name :: <==> {item}")
-                fp.write("\n")
+                fp.write("\n")
+
+import copy
+from transformers import AutoProcessor, AutoModelForCausalLM
+from PIL import Image
+import requests
+import utils
+
+
+model_id = 'microsoft/Florence-2-large'
+model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True).eval().cuda()
+processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
+
+utils.set_model_info(model, processor)
+
+from google.colab import drive
+drive.mount('/content/drive')
+
+path = "/content/pci complliance and Credit card Authorization forms Hand written - Copy.jpg"
+
+image = Image.open(path)
+image_rgb = Image.open(path).convert("RGB")
+image
+
+img_list = ["/content/images_ocr_image_3.jpg",
+ "/content/images_ocr_image_2.jpg",
+ "/content/images_ocr_image_1.jpg",
+ "/content/images_ocr_image_4.png",
+ "/content/pci complliance and Credit card Authorization forms Hand written - Copy.jpg",
+ "/content/Charlemagne.png",
+ "/content/Frederick II.png",
+ "/content/Henry VIII.png",
+ "/content/Louis XIV.png",
+ "/content/William IV.png",
+ "/content/pci complliance and Credit card Authorization forms Hand written - Copy.jpg"
+]
+for item in img_list:
+  path = item
+  image = Image.open(path)
+  image_rgb = Image.open(path).convert("RGB")
+
+  tasks = [utils.TaskType.CAPTION,
+         utils.TaskType.DETAILED_CAPTION,
+         utils.TaskType.MORE_DETAILED_CAPTION,]
+
+  for task in tasks:
+    results = utils.run_example(task, image_rgb)
+    print(f'{task.value}{results[task]}')
+
+  task = utils.TaskType.OCR
+  results = utils.run_example(task, image_rgb)
+  print('Detected Text: ', results[task])
+
+  with open("/content/florenecre.txt", "a+" ) as fp:
+    fp.write(results[task])
+    fp.write("\n")
+    fp.write("Image Name :: <==>",item,"==="*30,item)
+    fp.write("\n")