Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from transformers import pipeline | |
| # Load the model | |
| qa_pipeline = pipeline("text-generation", model="EleutherAI/gpt-neo-125m") | |
| # Streamlit interface | |
| st.title("GPT-Neo 125M Q&A App") | |
| # Example questions | |
| example_questions = [ | |
| "Who is Spider-Man?", | |
| "Who is Venom?", | |
| "Who is OpenAI?", | |
| "Who is Rocket Raccoon?" | |
| ] | |
| st.subheader("Select an example question or type your own:") | |
| selected_question = st.selectbox("Choose an example question", example_questions) | |
| user_question = st.text_input("Or ask your own question:", value=selected_question) | |
| # Generate answer | |
| if user_question: | |
| with st.spinner('Generating answer...'): | |
| response = qa_pipeline(user_question, max_length=50) | |
| answer = response[0]['generated_text'] | |
| st.write(answer) | |