SchoolCal_st / src /streamlit_app.py
yongyeol's picture
Update src/streamlit_app.py
d0ecfa7 verified
raw
history blame
426 Bytes
import streamlit as st
from transformers import pipeline
st.set_page_config(page_title="pipeline ν…ŒμŠ€νŠΈ")
st.title("βœ… transformers.pipeline import 확인")
try:
pipe = pipeline("text-generation", model="gpt2")
output = pipe("Hello, my name is", max_new_tokens=10)
st.success("pipeline import 및 μ‹€ν–‰ 성곡!")
st.write(output[0]["generated_text"])
except Exception as e:
st.error(f"μ‹€νŒ¨: {e}")