app.py

import os
import streamlit as st
from transformers import pipeline

st.title("Testing Gemma")
input_text = st.text_input("How can i help you: ")
process = st.button("ask")

messages = [
    {"role": "user", "content": input_text},
]
pipe = pipeline("text-generation", model="google/gemma-3-1b-it",token=os.getenv("HF_TOKEN"))

if process:
    st.write(pipe(messages))