Spaces:
Running
on
Zero
Running
on
Zero
Enable logged-in quotas on ZeroGPU
Browse files
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],
|