jiuface commited on
Commit
a9452f7
1 Parent(s): a236279

Update utils/florence.py

Browse files
Files changed (1) hide show
  1. utils/florence.py +0 -2
utils/florence.py CHANGED
@@ -49,7 +49,6 @@ def run_florence_inference(
49
  else:
50
  prompt = task
51
  inputs = processor(text=prompt, images=image, return_tensors="pt").to(device)
52
- print(inputs)
53
  generated_ids = model.generate(
54
  input_ids=inputs["input_ids"],
55
  pixel_values=inputs["pixel_values"],
@@ -60,5 +59,4 @@ def run_florence_inference(
60
  generated_ids, skip_special_tokens=False)[0]
61
  response = processor.post_process_generation(
62
  generated_text, task=task, image_size=image.size)
63
- print(generated_text, response)
64
  return generated_text, response
 
49
  else:
50
  prompt = task
51
  inputs = processor(text=prompt, images=image, return_tensors="pt").to(device)
 
52
  generated_ids = model.generate(
53
  input_ids=inputs["input_ids"],
54
  pixel_values=inputs["pixel_values"],
 
59
  generated_ids, skip_special_tokens=False)[0]
60
  response = processor.post_process_generation(
61
  generated_text, task=task, image_size=image.size)
 
62
  return generated_text, response