Update app.py

app.py

@@ -1,24 +1,24 @@
 import streamlit as st
 from transformers import pipeline
 
-# Use a pipeline as a high-level helper
-messages = [
-    {"role": "user", "content": "Who are you?"}
+# Load the text-generation pipeline
 pipe = pipeline("text-generation", model="Leo022/Gemma_QA_For_Telegram_Bot")
 
 # Create a Streamlit app
-st.title("Gemma_QA_For_Telegram_Bot")
+st.title("Gemma QA for Telegram Bot")
 
 # Create a text input for the user to enter their question
-question = st.text_input("Enter your question")
+question = st.text_input("Enter your question:")
 
 # Create a button to submit the question
 if st.button("Submit"):
-    # Use the pipeline to generate a response to the user's question
-    response = pipe([{"role": "user", "content": question}])
-    # Display the response to the user
-    st.write("Response:", response[0]["generated_text"])
-
-# Run the app
-if __name__ == "__main__":
-    st.run()
+    if question.strip():  # Ensure the question is not empty
+        try:
+            # Use the pipeline to generate a response to the user's question
+            response = pipe(question, max_length=100, num_return_sequences=1)
+            # Display the response to the user
+            st.write("Response:", response[0]["generated_text"])
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
+    else:
+        st.warning("Please enter a question before submitting.")