Krog-v3 / app.py
IcedSundae's picture
Update app.py
4303cdd verified
raw
history blame contribute delete
396 Bytes
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()