import gradio as gr
import spaces
import transformers_gradio

# demo = gr.load(name="MakiAi/Llama-3-2-3B-Instruct-bnb-4bit-OKU-v1-10epochs-adapter", src=transformers_gradio.registry)
# demo = gr.load(name="unsloth/Llama-3.2-3B-Instruct-bnb-4bit", src=transformers_gradio.registry)
demo = gr.load(name="MakiAi/Llama-3-2-3B-Instruct-bnb-4bit-OKU-v1-1epochs", src=transformers_gradio.registry)

# demo = gr.load(name="unsloth/Llama-3.2-1B", src=transformers_gradio.registry)



demo.fn = spaces.GPU()(demo.fn)

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