llamameta's picture
Create test.py
388dac5 verified
raw
history blame contribute delete
285 Bytes
import gradio as gr
import spaces
import transformers_gradio
demo = gr.load(name="moonshotai/Kimi-K2-Instruct-0905", src=transformers_gradio.registry)
demo.fn = spaces.GPU()(demo.fn)
for fn in demo.fns.values():
fn.api_name = False
if __name__ == "__main__":
demo.launch()