app.py

"""
"""

# Upgrade PyTorch
import os
os.system('pip install --upgrade torch torchvision')

# CUDA toolkit install
from utils.cuda_toolkit import install_cuda_toolkit; install_cuda_toolkit()

# Try AOTI compile + load
import spaces
@spaces.GPU
def run():
    import aoti_base_example
run()

# Base demo
import gradio as gr

def greet(name):
    return "Hello " + name + "!!"

demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()