Update app.py

app.py

@@ -1,14 +1,11 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import tempfile
-
 # Corrected model class name
 model_name = "potsawee/t5-large-generation-squad-QuestionAnswer"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-
 uploaded_file = st.file_uploader("Upload Document or Paragraph")
-
 if uploaded_file is not None:
     with tempfile.NamedTemporaryFile(delete=False) as temp_file:
         temp_file.write(uploaded_file.read())
@@ -16,10 +13,8 @@ if uploaded_file is not None:
         st.success("Document uploaded successfully!")
 else:
     document_text = st.text_area("Enter Text (Optional)", height=200)
-
 question = st.text_input("Ask a Question")
 bouton_ok = st.button("Answer")
-
 if bouton_ok:
     # Improved prompt for better context
     context = document_text if document_text else "Empty document."