d6532bf
b4f0f17
d6532bf
b4f0f17
|
|
import os
import streamlit as st
from transformers import pipeline
messages = [
{"role": "user", "content": "Who are you?"},
]
pipe = pipeline("text-generation", model="google/gemma-3-1b-it",token=os.getenv("HF_TOKEN"))
st.write(pipe(messages)) |
