Update app.py

app.py

@@ -1,15 +1,27 @@
 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")
 
-# User input
-user_question = st.text_input("Ask a question:")
+# 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:
-    response = qa_pipeline(user_question, max_length=50)
-    answer = response[0]['generated_text']
-    st.write(answer)
+    with st.spinner('Generating answer...'):
+        response = qa_pipeline(user_question, max_length=50)
+        answer = response[0]['generated_text']
+        st.write(answer)