nolenfelten commited on
Commit
8b04a96
·
verified ·
1 Parent(s): 92bcee4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -273,7 +273,7 @@ def object_detection_density_edge(image, conf_threshold=0.25, iou_threshold=0.45
273
 
274
 
275
  # Function to resize and encode an image
276
- def resize_image(image, max_size=2500):
277
  print("resize_image initialized")
278
  if max(image.size) > max_size:
279
 
@@ -300,7 +300,7 @@ def resize_image(image, max_size=2500):
300
  return base64.b64encode(buffered).decode("utf-8")
301
 
302
 
303
- def infer(image, model, version, api_key, confidence=0.45, overlap=0.45, format="json", labels=False, stroke=1):
304
 
305
  base_url = f"https://detect.roboflow.com/{model}/{version}?api_key={api_key}&confidence={confidence}&overlap={overlap}&format={format}"
306
 
@@ -343,11 +343,11 @@ interface = gr.Interface(
343
  gr.Textbox(value="sku-110k", label="Model Name"),
344
  gr.Textbox(value="2", label="Model Version"),
345
  gr.Textbox(value="gHiUgOSq9GqTnRy5mErk", label="API Key"),
346
- gr.Slider(0.0, 1.0, value=0.45, label="Confidence Threshold"),
347
- gr.Slider(0.0, 1.0, value=0.45, label="Overlap Threshold"),
348
  gr.Radio(["json", "image"], value="image", label="Output Format"),
349
  gr.Checkbox(False, label="Include Labels"),
350
- gr.Slider(1, 10, value=1, step=1, label="Stroke Width"),
351
  ],
352
  outputs=[
353
  gr.Image(label="Roboflow Output Image"),
 
273
 
274
 
275
  # Function to resize and encode an image
276
+ def resize_image(image, max_size=1500):
277
  print("resize_image initialized")
278
  if max(image.size) > max_size:
279
 
 
300
  return base64.b64encode(buffered).decode("utf-8")
301
 
302
 
303
+ def infer(image, model, version, api_key, confidence=0.4, overlap=0.20, format="json", labels=False, stroke=1):
304
 
305
  base_url = f"https://detect.roboflow.com/{model}/{version}?api_key={api_key}&confidence={confidence}&overlap={overlap}&format={format}"
306
 
 
343
  gr.Textbox(value="sku-110k", label="Model Name"),
344
  gr.Textbox(value="2", label="Model Version"),
345
  gr.Textbox(value="gHiUgOSq9GqTnRy5mErk", label="API Key"),
346
+ gr.Slider(0.0, 1.0, value=0.40, label="Confidence Threshold"),
347
+ gr.Slider(0.0, 1.0, value=0.20, label="Overlap Threshold"),
348
  gr.Radio(["json", "image"], value="image", label="Output Format"),
349
  gr.Checkbox(False, label="Include Labels"),
350
+ gr.Slider(1, 10, value=5, step=1, label="Stroke Width"),
351
  ],
352
  outputs=[
353
  gr.Image(label="Roboflow Output Image"),