Spaces:

Mountchicken
/

Rex-Omni

Running on Zero

Mountchicken commited on Oct 14

Commit

459f575

verified ·

1 Parent(s): 961c93e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -355,7 +355,7 @@ def get_task_prompt(
         else:
             return task_config.prompt_template.replace("{categories}", "objects")
 def run_inference(
     image,
     task_selection,

         else:
             return task_config.prompt_template.replace("{categories}", "objects")
+@spaces.GPU
 def run_inference(
     image,
     task_selection,