Update app.py
Browse files
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,
|