import gradio as gr
from langchain.llms import CTransformers

# Load model
llm = CTransformers(
    model="IcedSundae/Krog-v1",
    model_type="llama",
    config={"gpu_layers": 0}  # Change to GPU layers if using paid plan
)

def chat(message):
    response = llm(message)
    return response

# Create UI
iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="Krog")
iface.launch()