app.py

import gradio as gr
import spaces
import transformers_gradio

# Load Llama model
demo = gr.load(name="allenai/Llama-3.1-Tulu-3-8B", src=transformers_gradio.registry)
demo.fn = spaces.GPU()(demo.fn)

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