Musa07 commited on
Commit
9d92a6a
·
verified ·
1 Parent(s): 7e030fb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -19,6 +19,8 @@ It achieves the following results on the evaluation set:
19
  - Loss: 0.2107
20
 
21
  ### Inference Code versions
 
 
22
  from transformers import AutoProcessor, AutoModelForCausalLM
23
  import matplotlib.pyplot as plt
24
  import matplotlib.patches as patches
@@ -27,6 +29,7 @@ It achieves the following results on the evaluation set:
27
  processor = AutoProcessor.from_pretrained("Musa07/Florence-2-large-FormClassification-ft", trust_remote_code=True)
28
 
29
  def run_example(task_prompt, image, max_new_tokens=128):
 
30
  prompt = task_prompt
31
  inputs = processor(text=prompt, images=image, return_tensors="pt")
32
  generated_ids = model.generate(
 
19
  - Loss: 0.2107
20
 
21
  ### Inference Code versions
22
+
23
+ # Code
24
  from transformers import AutoProcessor, AutoModelForCausalLM
25
  import matplotlib.pyplot as plt
26
  import matplotlib.patches as patches
 
29
  processor = AutoProcessor.from_pretrained("Musa07/Florence-2-large-FormClassification-ft", trust_remote_code=True)
30
 
31
  def run_example(task_prompt, image, max_new_tokens=128):
32
+
33
  prompt = task_prompt
34
  inputs = processor(text=prompt, images=image, return_tensors="pt")
35
  generated_ids = model.generate(