NexusRaven2 / app.py
Tonic's picture
Update app.py
6d607e0
raw
history blame
896 Bytes
from transformers import pipeline
pipeline = pipeline(
"text-generation",
model="Nexusflow/NexusRaven-V2-13B",
torch_dtype="auto",
device_map="auto",
)
title="""# 🙋🏻‍♂️Welcome to🌟Tonic's🐦‍⬛NexusRaven"""
description="""this model is used to select and return function calling arguments.
"""
prompt = prompt_template.format(query="What's the weather like in Seattle right now?")
result = pipeline(prompt, max_new_tokens=2048, return_full_text=False, do_sample=False, temperature=0.001)[0]["generated_text"]
print (result)
with gr.Blocks() as app:
gr.Markdown(title)
gr.Markdown(description)
with gr.Row():
with gr.Column():
input = gr.Textbox()
with gr.Column():
output = gr.Textbox()
image_input.change(process_image, inputs=image_input, outputs=output)
if __name__ == "__main__":
app.launch()