cbensimon HF staff commited on
Commit
32265ff
·
verified ·
1 Parent(s): 621be96

Enable logged-in quotas on ZeroGPU

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -234,6 +234,8 @@ def app():
234
  }
235
  return yoloe_inference(image, prompts, target_image, model_id, image_size, conf_thresh, iou_thresh, prompt_type)
236
 
 
 
237
  yoloe_infer.click(
238
  fn=run_inference,
239
  inputs=[raw_image, box_image, mask_image, target_image, texts, model_id, image_size, conf_thresh, iou_thresh, prompt_type, visual_prompt_type, visual_usage_type],
 
234
  }
235
  return yoloe_inference(image, prompts, target_image, model_id, image_size, conf_thresh, iou_thresh, prompt_type)
236
 
237
+ run_inference.zerogpu = True
238
+
239
  yoloe_infer.click(
240
  fn=run_inference,
241
  inputs=[raw_image, box_image, mask_image, target_image, texts, model_id, image_size, conf_thresh, iou_thresh, prompt_type, visual_prompt_type, visual_usage_type],