vumichien commited on
Commit
ab921ed
1 Parent(s): b2db8cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -23,6 +23,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
23
  model = AutoModelForCausalLM.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True).to(device).eval()
24
  processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
25
  onnx_model = YOLO("models/best.onnx", task='detect')
 
26
 
27
 
28
  def filter_detections(detections, target_class_name="mark"):
 
23
  model = AutoModelForCausalLM.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True).to(device).eval()
24
  processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base-ft", trust_remote_code=True)
25
  onnx_model = YOLO("models/best.onnx", task='detect')
26
+ onnx_model_table = YOLO("models/tables/best.onnx", task='detect')
27
 
28
 
29
  def filter_detections(detections, target_class_name="mark"):