Spaces:
Running
on
Zero
Running
on
Zero
File size: 450 Bytes
8c2e0d0 ee7158f 8c2e0d0 ee7158f 8c2e0d0 2e8adf4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
"""
"""
# 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()
|