File size: 229 Bytes
06128b2
59cb807
06128b2
 
 
59cb807
06128b2
 
 
1
2
3
4
5
6
7
8
9
import gradio as gr
import spaces
import transformers_gradio

demo = gr.load(name="allenai/OLMo-2-1124-13B-Instruct", src=transformers_gradio.registry)
demo.fn = spaces.GPU()(demo.fn)

if __name__ == "__main__":
    demo.launch()