Add ZeroGPU support

app.py

@@ -4,6 +4,7 @@ import json
 from pathlib import Path
 from typing import Optional, Tuple
 import gradio as gr
+import spaces
 from gradio_huggingfacehub_search import HuggingfaceHubSearch
 import huggingface_hub
 from sentence_transformers import CrossEncoder, SentenceTransformer, SparseEncoder
@@ -150,6 +151,7 @@ def export_to_torch(model_id, create_pr, output_model_id):
     )
 
 
+@spaces.GPU(duration=180)
 def export_to_onnx(
     model_id: str,
     archetype: Archetype,
@@ -287,6 +289,7 @@ predictions = model.predict([
     )
 
 
+@spaces.GPU(duration=180)
 def export_to_onnx_dynamic_quantization(
     model_id: str,
     archetype: Archetype,
@@ -420,6 +423,7 @@ predictions = model.predict([
     )
 
 
+@spaces.GPU(duration=180)
 def export_to_onnx_optimization(
     model_id: str,
     archetype: Archetype,
@@ -530,6 +534,7 @@ predictions = model.predict([
     )
 
 
+@spaces.GPU(duration=180)
 def export_to_openvino(
     model_id: str,
     archetype: Archetype,
@@ -665,6 +670,7 @@ predictions = model.predict([
     )
 
 
+@spaces.GPU(duration=300)
 def export_to_openvino_static_quantization(
     model_id: str,
     archetype: Archetype,

requirements.txt

@@ -1,3 +1,4 @@
+spaces==0.50.2
 sentence_transformers[onnx-gpu,openvino]==5.1.0
 onnx==1.16.1
 https://huggingface.co/spaces/CISCai/chat-template-editor/resolve/08c8e90c53677ae70c66b3d90bf4e63a173b5505/gradio_huggingfacehub_search-0.0.8-py3-none-any.whl