Upload gradio_app.py

gradio_app.py

@@ -31,7 +31,7 @@ from hy3dgen.shapegen.utils import logger
 
 MAX_SEED = 1e7
 
-import spaces
+from gradio.decorators import spaces
 import subprocess
 import sys
 import shlex
@@ -41,8 +41,9 @@ print('install custom')
 subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
 
 @spaces.GPU
-def dummy_function():
-    pass
+def my_gpu_function(text):
+    # Your GPU-enabled code here
+    return text + " processed on GPU"
 
 def get_example_img_list():
     print('Loading example img list ...')
@@ -362,7 +363,6 @@ def shape_generation(
     )
 
 
-@spaces.GPU
 def build_app():
     title = 'Hunyuan3D-2: High Resolution Textured 3D Assets Generation'
     if MV_MODE:
@@ -658,7 +658,7 @@ def build_app():
 if __name__ == '__main__':
     import argparse
 
-    dummy_function()
+   gr.Interface(fn=my_gpu_function, inputs="text", outputs="text")
 
     parser = argparse.ArgumentParser()
     parser.add_argument("--model_path", type=str, default='tencent/Hunyuan3D-2mv')