capjamesg commited on
Commit
76eb2df
·
verified ·
1 Parent(s): ebcf14b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ CLIENT = InferenceHTTPClient(
16
  api_key=os.environ["ROBOFLOW_API_KEY"],
17
  )
18
 
19
- custom_configuration = InferenceConfiguration(confidence_threshold=0.3)
20
 
21
  openai_client = OpenAI()
22
 
@@ -85,7 +85,7 @@ def process_book_with_google_books(book):
85
  def detect_books(image):
86
  # infer on a local image
87
  with CLIENT.use_configuration(custom_configuration):
88
- results = CLIENT.infer(image, model_id="open-shelves/6")
89
  results = sv.Detections.from_inference(results)
90
 
91
  mask_annotator = sv.MaskAnnotator()
 
16
  api_key=os.environ["ROBOFLOW_API_KEY"],
17
  )
18
 
19
+ custom_configuration = InferenceConfiguration(confidence_threshold=0.2)
20
 
21
  openai_client = OpenAI()
22
 
 
85
  def detect_books(image):
86
  # infer on a local image
87
  with CLIENT.use_configuration(custom_configuration):
88
+ results = CLIENT.infer(image, model_id="open-shelves/8")
89
  results = sv.Detections.from_inference(results)
90
 
91
  mask_annotator = sv.MaskAnnotator()