ST-x-Tony commited on
Commit
c2e0682
·
verified ·
1 Parent(s): 60c933d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -12,6 +12,7 @@ import asyncio
12
  import traceback
13
 
14
  import gradio as gr
 
15
 
16
 
17
  # ============================================================
@@ -652,9 +653,10 @@ async def do_research(
652
 
653
 
654
  # ============================================================
655
- # GRADIO SYNC WRAPPER
656
  # ============================================================
657
 
 
658
  def run_research(
659
  question,
660
  max_results,
@@ -662,7 +664,6 @@ def run_research(
662
  use_models,
663
  freshness,
664
  ):
665
-
666
  return asyncio.run(
667
  do_research(
668
  question,
 
12
  import traceback
13
 
14
  import gradio as gr
15
+ import spaces # <-- ADDED: required for ZeroGPU
16
 
17
 
18
  # ============================================================
 
653
 
654
 
655
  # ============================================================
656
+ # GRADIO SYNC WRAPPER – DECORATED FOR ZEROGPU
657
  # ============================================================
658
 
659
+ @spaces.GPU # <-- ADDED: required for ZeroGPU allocation
660
  def run_research(
661
  question,
662
  max_results,
 
664
  use_models,
665
  freshness,
666
  ):
 
667
  return asyncio.run(
668
  do_research(
669
  question,