Update app.py

app.py

@@ -1,4 +1,3 @@
-import torch
 import streamlit as st
 from transformers import pipeline
 
@@ -6,11 +5,11 @@ st.set_page_config(page_title="Common NLP Tasks")
 st.title("Common NLP Tasks")
 st.subheader("Use the menu on the left to select a NLP task to do (click on > if closed).")
 
-expander = st.sidebar.expander('About')
+expander = st.sidebar.expander("About")
 expander.write("This web app allows you to perform common Natural Language Processing tasks, select a task below to get started.")
 
-st.sidebar.header('What will you like to do?')
-option = st.sidebar.radio('', ['Extractive question answering', 'Text summarization', 'Text generation', 'Sentiment analysis'])
+st.sidebar.header("What will you like to do?")
+option = st.sidebar.radio("", ["Extractive question answering", "Text summarization", "Text generation", "Sentiment analysis"])
 
 @st.cache(show_spinner=False, allow_output_mutation=True)
 def question_model():
@@ -33,7 +32,7 @@ def sentiment_model():
     sentiment_analysis = pipeline("sentiment-analysis")
     return sentiment_analysis
 
-if option == 'Extractive question answering':
+if option == "Extractive question answering":
     st.markdown("<h2 style='text-align: center; color:red;'>Extract answer from text</h2>", unsafe_allow_html=True)
     source = st.radio("How would you like to start? Choose an option below", ["I want to input some text", "I want to upload a file"])
     if source == "I want to input some text":
@@ -46,7 +45,7 @@ if option == 'Extractive question answering':
             question_answerer = question_model()
             with st.spinner(text="Getting answer..."):
                 answer = question_answerer(context=context, question=question)
-                st.write(answer["answer"])
+                st.write(f"Answer: {answer["answer"]}")
     elif source == "I want to upload a file":
         uploaded_file = st.file_uploader("Choose a .txt file to upload", type=["txt"])
         question = st.text_input(label='Enter your question')
@@ -54,8 +53,8 @@ if option == 'Extractive question answering':
         if button:
             question_answerer = question_model()
             with st.spinner(text="Getting answer..."):
-                answer = question_answerer(context=context, question=question)
-                st.write(answer["answer"])
+                answer = question_answerer(context=uploaded_file, question=question)
+                st.write(f"Answer: {answer["answer"]}")
 
 elif option == 'Text summarization':
     st.markdown("<h2 style='text-align: center; color:red;'>Summarize text</h2>", unsafe_allow_html=True)