import streamlit as st | |
from transformers import pipeline | |
messages = [ | |
{"role": "user", "content": "Who are you?"}, | |
] | |
pipe = pipeline("text-generation", model="iGeniusAI/Italia-9B-Instruct-v0.1", trust_remote_code=True) | |
pipe(messages) | |
#pipe = pipeline('sentiment-analysis') | |
#text =st.text_area('dsfs') | |
#if text: | |
out = pipe(messages) | |
st.json(out) |