processing_test / app.py
Mori-kamiyama's picture
Pytorchからllama.cppに変更
f251f9b
raw
history blame
444 Bytes
import gradio as gr
from llama_cpp import Llama
# gguf形式のモデルファイルのパスを指定してロード
llm = Llama(model_path="Mori-kamiyama/sarashina2-13b-r1/model.gguf")
def generate_text(prompt):
return llm(prompt)
iface = gr.Interface(fn=generate_text,
inputs="text",
outputs="text",
title="GGUFモデルテキストジェネレーター")
iface.launch()